diff --git a/cli/src/generate/dsl.js b/cli/src/generate/dsl.js
index fc640a69..d4ef2f25 100644
--- a/cli/src/generate/dsl.js
+++ b/cli/src/generate/dsl.js
@@ -229,6 +229,8 @@ function grammar(baseGrammar, options) {
       supertypes: [],
       precedences: [],
     };
+  } else {
+    baseGrammar = baseGrammar.grammar;
   }
 
   let externals = baseGrammar.externals;