diff --git a/ace-loader/src/loader-gen.js b/ace-loader/src/loader-gen.js
index f61c940..dde5da4 100644
--- a/ace-loader/src/loader-gen.js
+++ b/ace-loader/src/loader-gen.js
@@ -211,7 +211,7 @@ function codegenHmlAndCss() {
   return output
 }
 
-function generateOutput(that, type, jsFileName, isElement) {
+function generateOutput(that, type, jsFileName, isElement, customLang) {
   let result
   switch (type) {
     case 'script':