diff --git a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/ControllerGenerator.java b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/ControllerGenerator.java
index ec59a22d..63c52008 100644
--- a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/ControllerGenerator.java
+++ b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/ControllerGenerator.java
@@ -71,7 +71,7 @@ public class ControllerGenerator implements IGenerator {
         params.put("javadocConfig", globalConfig.getJavadocConfig());
         params.put("withSwagger", globalConfig.isEntityWithSwagger());
         params.put("swaggerVersion", globalConfig.getSwaggerVersion());
-        params.putAll(globalConfig.getCustomConfig());
+        params.put("customConfig", globalConfig.getCustomConfig());
         globalConfig.getTemplateConfig().getTemplate().generate(params, templatePath, controllerJavaFile);
 
         System.out.println("Controller ---> " + controllerJavaFile);
diff --git a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/EntityGenerator.java b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/EntityGenerator.java
index e4b12817..927dff3f 100644
--- a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/EntityGenerator.java
+++ b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/EntityGenerator.java
@@ -72,7 +72,7 @@ public class EntityGenerator implements IGenerator {
         params.put("entityConfig", entityConfig);
         params.put("packageConfig", packageConfig);
         params.put("javadocConfig", globalConfig.getJavadocConfig());
-        params.putAll(globalConfig.getCustomConfig());
+        params.put("customConfig", globalConfig.getCustomConfig());
         globalConfig.getTemplateConfig().getTemplate().generate(params, templatePath, entityJavaFile);
 
         System.out.println("Entity ---> " + entityJavaFile);
diff --git a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/MapperGenerator.java b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/MapperGenerator.java
index 12bd7d61..143bbfb5 100644
--- a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/MapperGenerator.java
+++ b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/MapperGenerator.java
@@ -69,7 +69,7 @@ public class MapperGenerator implements IGenerator {
         params.put("mapperConfig", mapperConfig);
         params.put("packageConfig", packageConfig);
         params.put("javadocConfig", globalConfig.getJavadocConfig());
-        params.putAll(globalConfig.getCustomConfig());
+        params.put("customConfig", globalConfig.getCustomConfig());
         globalConfig.getTemplateConfig().getTemplate().generate(params, templatePath, mapperJavaFile);
 
         System.out.println("Mapper ---> " + mapperJavaFile);
diff --git a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/MapperXmlGenerator.java b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/MapperXmlGenerator.java
index ad6fbad3..9d314964 100644
--- a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/MapperXmlGenerator.java
+++ b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/MapperXmlGenerator.java
@@ -66,7 +66,7 @@ public class MapperXmlGenerator implements IGenerator {
         Map<String, Object> params = new HashMap<>(2);
         params.put("table", table);
         params.put("packageConfig", packageConfig);
-        params.putAll(globalConfig.getCustomConfig());
+        params.put("customConfig", globalConfig.getCustomConfig());
         globalConfig.getTemplateConfig().getTemplate().generate(params, templatePath, mapperXmlFile);
 
         System.out.println("MapperXML ---> " + mapperXmlFile);
diff --git a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/ServiceGenerator.java b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/ServiceGenerator.java
index 3a7ce530..1eedd5a2 100644
--- a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/ServiceGenerator.java
+++ b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/ServiceGenerator.java
@@ -69,7 +69,7 @@ public class ServiceGenerator implements IGenerator {
         params.put("serviceConfig", serviceConfig);
         params.put("packageConfig", packageConfig);
         params.put("javadocConfig", globalConfig.getJavadocConfig());
-        params.putAll(globalConfig.getCustomConfig());
+        params.put("customConfig", globalConfig.getCustomConfig());
         globalConfig.getTemplateConfig().getTemplate().generate(params, templatePath, serviceJavaFile);
 
         System.out.println("Service ---> " + serviceJavaFile);
diff --git a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/ServiceImplGenerator.java b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/ServiceImplGenerator.java
index 1d9a74a0..a465dffc 100644
--- a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/ServiceImplGenerator.java
+++ b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/ServiceImplGenerator.java
@@ -69,7 +69,7 @@ public class ServiceImplGenerator implements IGenerator {
         params.put("packageConfig", packageConfig);
         params.put("serviceImplConfig", serviceImplConfig);
         params.put("javadocConfig", globalConfig.getJavadocConfig());
-        params.putAll(globalConfig.getCustomConfig());
+        params.put("customConfig", globalConfig.getCustomConfig());
         globalConfig.getTemplateConfig().getTemplate().generate(params, templatePath, serviceImplJavaFile);
 
         System.out.println("ServiceImpl ---> " + serviceImplJavaFile);
diff --git a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/TableDefGenerator.java b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/TableDefGenerator.java
index 420c6c2a..0ebca1aa 100644
--- a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/TableDefGenerator.java
+++ b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/TableDefGenerator.java
@@ -69,7 +69,7 @@ public class TableDefGenerator implements IGenerator {
         params.put("packageConfig", packageConfig);
         params.put("tableDefConfig", tableDefConfig);
         params.put("javadocConfig", globalConfig.getJavadocConfig());
-        params.putAll(globalConfig.getCustomConfig());
+        params.put("customConfig", globalConfig.getCustomConfig());
         globalConfig.getTemplateConfig().getTemplate().generate(params, templatePath, tableDefJavaFile);
 
         System.out.println("TableDef ---> " + tableDefJavaFile);