diff --git a/mybatis-flex-codegen/src/test/java/com/mybatisflex/codegen/test/GeneratorTest.java b/mybatis-flex-codegen/src/test/java/com/mybatisflex/codegen/test/GeneratorTest.java
index b2ce21e1..cf0155a5 100644
--- a/mybatis-flex-codegen/src/test/java/com/mybatisflex/codegen/test/GeneratorTest.java
+++ b/mybatis-flex-codegen/src/test/java/com/mybatisflex/codegen/test/GeneratorTest.java
@@ -85,7 +85,7 @@ public class GeneratorTest {
         generator.generate();
     }
 
-    //        @Test
+    //    @Test
     public void testCodeGen2() {
         //配置数据源
         HikariDataSource dataSource = new HikariDataSource();
@@ -122,6 +122,7 @@ public class GeneratorTest {
         globalConfig.enableEntity()
             .setOverwriteEnable(true)
             .setWithLombok(true)
+            .setWithSwagger(true)
             .setSuperClass(BaseEntity.class);
 
         //配置生成 mapper