diff --git a/jcommon/codegen/src/test/java/com/xiaomi/youpin/codegen/test/FeatureGeneratorTest.java b/jcommon/codegen/src/test/java/com/xiaomi/youpin/codegen/test/FeatureGeneratorTest.java
index ff39ead0e..13614e766 100644
--- a/jcommon/codegen/src/test/java/com/xiaomi/youpin/codegen/test/FeatureGeneratorTest.java
+++ b/jcommon/codegen/src/test/java/com/xiaomi/youpin/codegen/test/FeatureGeneratorTest.java
@@ -5,6 +5,7 @@
 import run.mone.ai.codegen.MybatisGenerator;
 import run.mone.ai.codegen.bo.FeatureGeneratType;
 import run.mone.ai.codegen.bo.FeatureGenerateBo;
+import run.mone.ai.codegen.bo.FeatureGenerateType;
 import run.mone.ai.codegen.bo.MybatisGeneratorResult;
 
 public class FeatureGeneratorTest {
@@ -16,7 +17,7 @@ public class FeatureGeneratorTest {
 	@Test
 	public void testCreateTable() {
 	    FeatureGenerateBo featureGenerateBo = new FeatureGenerateBo();
-	    featureGenerateBo.setType(FeatureGeneratType.TABLE);
+	    featureGenerateBo.setType(FeatureGenerateType.TABLE);
 	    featureGenerateBo.setJdbcUrl(dbUrl);
 	    featureGenerateBo.setUserName(dbUser);
 	    featureGenerateBo.setPassword(dbPwd);
@@ -27,13 +28,12 @@ public void testCreateTable() {
 				"  PRIMARY KEY (`id`)\n" +
 				");";
 		featureGenerateBo.setSql(sql);
-		FeatureGenerator.generateWithTemplate(featureGenerateBo);
+		FeatureGenerator.generateWithGenerator(featureGenerateBo);
 	}
 
 	@Test
 	public void testGenerateWithMybatis() {
 	    FeatureGenerateBo featureGenerateBo = new FeatureGenerateBo();
-	    featureGenerateBo.setType(FeatureGeneratType.CODE_WITH_MYBATIS_GENERATOR);
 		featureGenerateBo.setJdbcUrl(dbUrl);
 		featureGenerateBo.setUserName(dbUser);
 		featureGenerateBo.setPassword(dbPwd);