diff --git a/src/frontend/src/stores/flowStore.ts b/src/frontend/src/stores/flowStore.ts
index 85c7ec60f..7f5d500f3 100644
--- a/src/frontend/src/stores/flowStore.ts
+++ b/src/frontend/src/stores/flowStore.ts
@@ -201,7 +201,7 @@ const useFlowStore = create<FlowStoreType>((set, get) => ({
         .map((node) => ({ ...node, selected: false }))
         .concat({ ...newNode, selected: false });
     });
-    set({ nodes: newNodes });
+    get().setNodes(newNodes);
 
     selection.edges.forEach((edge: Edge) => {
       let source = idsMap[edge.source];
@@ -245,7 +245,7 @@ const useFlowStore = create<FlowStoreType>((set, get) => ({
         newEdges.map((edge) => ({ ...edge, selected: false }))
       );
     });
-    set({ edges: newEdges });
+    get().setEdges(newEdges);
   },
   setLastCopiedSelection: (newSelection) => {
     set({ lastCopiedSelection: newSelection });