From 16e959b6393f61be44ee94aecfef1eabe198fca4 Mon Sep 17 00:00:00 2001
From: Lucas Oliveira <lucas.edu.oli@hotmail.com>
Date: Wed, 24 Jan 2024 18:57:47 +0100
Subject: [PATCH] Fixed not saving on drop

---
 src/frontend/src/stores/flowStore.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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