diff --git a/legend-engine-core/legend-engine-core-base/legend-engine-core-language-pure/legend-engine-language-pure-grammar-http-api/src/test/java/org/finos/legend/engine/language/pure/grammar/api/test/TestGrammarModelApi.java b/legend-engine-core/legend-engine-core-base/legend-engine-core-language-pure/legend-engine-language-pure-grammar-http-api/src/test/java/org/finos/legend/engine/language/pure/grammar/api/test/TestGrammarModelApi.java
index 1703b29c030..73887d76af9 100644
--- a/legend-engine-core/legend-engine-core-base/legend-engine-core-language-pure/legend-engine-language-pure-grammar-http-api/src/test/java/org/finos/legend/engine/language/pure/grammar/api/test/TestGrammarModelApi.java
+++ b/legend-engine-core/legend-engine-core-base/legend-engine-core-language-pure/legend-engine-language-pure-grammar-http-api/src/test/java/org/finos/legend/engine/language/pure/grammar/api/test/TestGrammarModelApi.java
@@ -68,13 +68,13 @@ public Class getBatchResultSpecializedClass()
     @Override
     public Function5<String, String, Integer, Integer, Boolean, Response> grammarToJson()
     {
-        return (a, b, c, d, e) -> grammarToJson.model(a, b, c, e, null);
+        return (a, b, c, d, e) -> grammarToJson.model(a, b, c, e, null, null);
     }
 
     @Override
     public Function2<PureModelContextData, RenderStyle, Response> jsonToGrammar()
     {
-        return (a, b) -> jsonToGrammar.model(a, b, null);
+        return (a, b) -> jsonToGrammar.model(a, b, null, null);
     }
 
     @Override