diff --git a/util/generate_models.dart b/util/generate_models.dart
index 78da9d214a9f924097a4634fe556a83444f12acf..24eb3077670141e4b9afbeefdcb41721ceec3e64 100644
--- a/util/generate_models.dart
+++ b/util/generate_models.dart
@@ -121,7 +121,7 @@ Future<String> getModel(String url) async {
     }
   }
   buffer.writeln('    });');
-  buffer.writeln('    return builder.build();');
+  buffer.writeln('    return builder.buildDocument();');
   buffer.writeln('  }');
   buffer.writeln('');