diff --git a/mybatis-flex-core/src/main/java/com/mybatisflex/core/util/UpdateEntity.java b/mybatis-flex-core/src/main/java/com/mybatisflex/core/util/UpdateEntity.java
index eb4eb1da..e76a0c57 100644
--- a/mybatis-flex-core/src/main/java/com/mybatisflex/core/util/UpdateEntity.java
+++ b/mybatis-flex-core/src/main/java/com/mybatisflex/core/util/UpdateEntity.java
@@ -32,11 +32,13 @@ public class UpdateEntity {
 
 
     public static <T> T of(Class<T> clazz) {
+        clazz = ClassUtil.getUsefulClass(clazz);
         return ModifyAttrsRecordProxyFactory.getInstance().get(clazz);
     }
 
 
     public static <T> T of(Class<T> clazz, Object id) {
+        clazz = ClassUtil.getUsefulClass(clazz);
         T newEntity = ModifyAttrsRecordProxyFactory.getInstance().get(clazz);
         TableInfo tableInfo = TableInfoFactory.ofEntityClass(clazz);
         List<IdInfo> primaryKeyList = tableInfo.getPrimaryKeyList();