diff --git a/yudao-module-infra/yudao-module-infra-server/src/test/java/cn/iocoder/yudao/module/infra/service/codegen/CodegenServiceImplTest.java b/yudao-module-infra/yudao-module-infra-server/src/test/java/cn/iocoder/yudao/module/infra/service/codegen/CodegenServiceImplTest.java
index 8ca4a0e3..c0293306 100644
--- a/yudao-module-infra/yudao-module-infra-server/src/test/java/cn/iocoder/yudao/module/infra/service/codegen/CodegenServiceImplTest.java
+++ b/yudao-module-infra/yudao-module-infra-server/src/test/java/cn/iocoder/yudao/module/infra/service/codegen/CodegenServiceImplTest.java
@@ -460,7 +460,7 @@ public class CodegenServiceImplTest extends BaseDbUnitTest {
             assertEquals(column01, columns.get(0));
             assertEquals(column02, columns.get(1));
             return true;
-        }), isNull(), isNull())).thenReturn(codes);
+        }), isNull(), isNull(), eq(false))).thenReturn(codes);
         // 准备参数
         Long tableId = table.getId();
 
@@ -509,7 +509,7 @@ public class CodegenServiceImplTest extends BaseDbUnitTest {
             assertEquals(1, columns.size());
             assertPojoEquals(subColumn01, columns.size());
             return true;
-        }))).thenReturn(codes);
+        }), eq(false))).thenReturn(codes);
         // 准备参数
         Long tableId = table.getId();