From 4cdd66d1d3e981a7625cdce7e70dc42264c74e39 Mon Sep 17 00:00:00 2001
From: anovazzi1 <otavio2204@gmail.com>
Date: Thu, 14 Sep 2023 20:22:29 -0300
Subject: [PATCH] fix(reactflowUtils.ts): add missing assignment of
 newSourceHandle to newEdge.data.sourceHandle to ensure data consistency
 fix(reactflowUtils.ts): add missing assignment of newTargetHandle to
 newEdge.data.targetHandle to ensure data consistency

---
 src/frontend/src/utils/reactflowUtils.ts | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/frontend/src/utils/reactflowUtils.ts b/src/frontend/src/utils/reactflowUtils.ts
index 43279aa33..755f4df59 100644
--- a/src/frontend/src/utils/reactflowUtils.ts
+++ b/src/frontend/src/utils/reactflowUtils.ts
@@ -731,6 +731,7 @@ export function ungroupNode(
             if(inputTypes){
               newTargetHandle.inputTypes = inputTypes;
             }
+            newEdge.data.targetHandle = newTargetHandle;
             newEdge.targetHandle = scapedJSONStringfy(newTargetHandle);
           }
         }
@@ -740,6 +741,7 @@ export function ungroupNode(
       newEdge.source = lastNode!.id;
       let newSourceHandle: sourceHandleType = scapeJSONParse(newEdge.sourceHandle!);
       newSourceHandle.id = lastNode!.id;
+      newEdge.data.sourceHandle = newSourceHandle;
       newEdge.sourceHandle = scapedJSONStringfy(newSourceHandle);
     }
     updatedEdges.push(newEdge);