diff --git a/packages/core/flow-engine/src/components/subModel/AddFieldButton.tsx b/packages/core/flow-engine/src/components/subModel/AddFieldButton.tsx
index 001a4afd83..551544e9bc 100644
--- a/packages/core/flow-engine/src/components/subModel/AddFieldButton.tsx
+++ b/packages/core/flow-engine/src/components/subModel/AddFieldButton.tsx
@@ -114,7 +114,8 @@ const AddFieldButtonCore: React.FC<AddFieldButtonProps> = ({
         if (fieldClass && fieldInterfaceName) {
           const defaultOptions = {
             ...fieldClass.meta?.defaultOptions,
-            use: fieldClass.name,
+            use:
+              model.flowEngine.findModelClass((name, ModelClass) => ModelClass === fieldClass)?.[0] || fieldClass.name,
           };
           const fieldItem = {
             key: field.name,