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