diff --git a/src/utils/convertFlowData.ts b/src/utils/convertFlowData.ts
index c464db2..6cd0236 100644
--- a/src/utils/convertFlowData.ts
+++ b/src/utils/convertFlowData.ts
@@ -77,7 +77,9 @@ export const convertFlowData = (flowData: any) => {
     if (nodeConfig.component) {
       node.data.component = {
         compIdentifier: nodeConfig.component.compIdentifier,
-        compInstanceIdentifier: nodeConfig.component.compInstanceIdentifier
+        compInstanceIdentifier: nodeConfig.component.compInstanceIdentifier,
+        customDef: nodeConfig.component?.customDef,
+        type: nodeConfig.component.type
       };
     }
 
@@ -156,6 +158,7 @@ export const revertFlowData = (nodes: any[], edges: any[]) => {
           compIdentifier: node.data.component.compIdentifier,
           compInstanceIdentifier: node.data.component.compInstanceIdentifier
         };
+        if (node.data.component?.customDef) nodeConfig.component.customDef = node.data.component.customDef;
       }
       else if (nodeType !== 'start' && nodeType !== 'end') {
         // 对于非 start/end 节点，添加基本的 component 信息