diff --git a/packages/generator/src/index.js b/packages/generator/src/index.js
index 6e420c7e35..a1e2bdff92 100644
--- a/packages/generator/src/index.js
+++ b/packages/generator/src/index.js
@@ -1,6 +1,6 @@
 import Generator from './generator'
 export { default as Generator } from './generator'
 
-export function getGenerator (nuxt) {
-  return new Generator(nuxt)
+export function getGenerator (nuxt, builder) {
+  return new Generator(nuxt, builder)
 }