diff --git a/docs/zh/others/codegen.md b/docs/zh/others/codegen.md
index 707868fe..3d8cddf9 100644
--- a/docs/zh/others/codegen.md
+++ b/docs/zh/others/codegen.md
@@ -79,7 +79,7 @@ public class Codegen {
         globalConfig.setEntityGenerateEnable(true);
         globalConfig.setEntityWithLombok(true);
         //设置项目的JDK版本，项目的JDK为14及以上时建议设置该项，小于14则可以不设置
-        globalConfig.setJdkVersion(17);
+        globalConfig.setEntityJdkVersion(17);
 
         //设置生成 mapper
         globalConfig.setMapperGenerateEnable(true);
diff --git a/mybatis-flex-core/src/main/java/com/mybatisflex/core/keygen/CustomKeyGenerator.java b/mybatis-flex-core/src/main/java/com/mybatisflex/core/keygen/CustomKeyGenerator.java
index da46f386..669ae0d1 100644
--- a/mybatis-flex-core/src/main/java/com/mybatisflex/core/keygen/CustomKeyGenerator.java
+++ b/mybatis-flex-core/src/main/java/com/mybatisflex/core/keygen/CustomKeyGenerator.java
@@ -80,8 +80,10 @@ public class CustomKeyGenerator implements KeyGenerator {
                 MetaObject metaParam = configuration.newMetaObject(parameter);
                 Object generateId = keyGenerator.generate(entity, idInfo.getColumn());
                 MetaObject metaObjectForProperty = metaParam.metaObjectForProperty(FlexConsts.ENTITY);
-                Invoker setInvoker = tableInfo.getReflector().getSetInvoker(idInfo.getProperty());
-                Object id = ConvertUtil.convert(generateId, setInvoker.getType());
+                // Invoker setInvoker = tableInfo.getReflector().getSetInvoker(idInfo.getProperty());
+                // Object id = ConvertUtil.convert(generateId, setInvoker.getType());
+                Class<?> setterType = tableInfo.getReflector().getSetterType(idInfo.getProperty());
+                Object id = ConvertUtil.convert(generateId, setterType);
                 this.setValue(metaObjectForProperty, this.idInfo.getProperty(), id);
             }
         } catch (Exception e) {