From 6d98841d65257be8678007970754fec3a298c123 Mon Sep 17 00:00:00 2001
From: michael <fuhai999@gmail.com>
Date: Wed, 13 Dec 2023 16:12:40 +0800
Subject: [PATCH] fix: fixed issues in gitee #I8NF9T

---
 .../src/main/java/com/mybatisflex/core/util/UpdateEntity.java   | 2 ++
 1 file changed, 2 insertions(+)

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();