From bb1a97d7a8cb7375c18c48567fa544520f5e5bd7 Mon Sep 17 00:00:00 2001
From: peterkogo <peter@xyflow.com>
Date: Tue, 25 Mar 2025 13:43:30 +0100
Subject: [PATCH] fix uncontrolled flows

---
 packages/react/src/store/index.ts | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/packages/react/src/store/index.ts b/packages/react/src/store/index.ts
index 8b219d27..e099ca95 100644
--- a/packages/react/src/store/index.ts
+++ b/packages/react/src/store/index.ts
@@ -64,7 +64,6 @@ const createStore = ({
         fitViewResolver?.resolve(value);
         set({ fitViewResolver: null });
       });
-      set({ nodes, fitViewQueued: false, fitViewOptions: undefined });
     }
 
     return {
@@ -89,6 +88,7 @@ const createStore = ({
 
         if (fitViewQueued && nodesInitialized) {
           resolveFitView();
+          set({ nodes, fitViewQueued: false, fitViewOptions: undefined });
         } else {
           set({ nodes });
         }
@@ -138,6 +138,7 @@ const createStore = ({
 
         if (fitViewQueued) {
           resolveFitView();
+          set({ fitViewQueued: false, fitViewOptions: undefined });
         } else {
           // we always want to trigger useStore calls whenever updateNodeInternals is called
           set({});