diff --git a/src/components/Edges/CustomEdge.tsx b/src/components/Edges/CustomEdge.tsx
index a916e5d..3bdc47e 100644
--- a/src/components/Edges/CustomEdge.tsx
+++ b/src/components/Edges/CustomEdge.tsx
@@ -126,7 +126,7 @@ const CustomEdge = ({
       Math.pow(t, 3) * params.ty;
 
     return { edgePath, labelX, labelY };
-  }, [sourceNode, targetNode, sourceShape, targetShape, sourceX, sourceY, targetX, targetY, data]);
+  }, [sourceNode, targetNode, sourceShape, targetShape, sourceX, sourceY, targetX, targetY, data, source, target]);
 
   const { edgePath, labelX, labelY } = edgeParams;
 
diff --git a/src/components/Editor/GraphEditor.tsx b/src/components/Editor/GraphEditor.tsx
index 7467751..d28f374 100644
--- a/src/components/Editor/GraphEditor.tsx
+++ b/src/components/Editor/GraphEditor.tsx
@@ -776,7 +776,6 @@ const GraphEditor = ({ presentationMode = false, onNodeSelect, onEdgeSelect, onG
       addEdgeWithHistory(newEdge);
     },
     [
-      storeEdges,
       edgeTypeConfigs,
       addEdgeWithHistory,
       selectedRelationType,