diff --git a/src/utils/graph.ts b/src/utils/graph.ts
index 433aad57..d48405bf 100644
--- a/src/utils/graph.ts
+++ b/src/utils/graph.ts
@@ -136,7 +136,7 @@ export const updateEdge = (oldEdge: Edge, newConnection: Connection, elements: E
     sourceHandle: newConnection.sourceHandle,
     targetHandle: newConnection.targetHandle,
   }
-
+  elements.splice(elements.indexOf(foundEdge), 1, edge)
   return elements.filter((e) => e.id !== oldEdge.id).concat(edge)
 }