From 0502b7069e3c563c8ffde02b7031844f56da4e4e Mon Sep 17 00:00:00 2001
From: peterkogo <peter@xyflow.com>
Date: Thu, 22 Aug 2024 17:11:39 +0200
Subject: [PATCH] fix setNodeExtent

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

diff --git a/packages/react/src/store/index.ts b/packages/react/src/store/index.ts
index 2941e89f..d70c9c45 100644
--- a/packages/react/src/store/index.ts
+++ b/packages/react/src/store/index.ts
@@ -288,10 +288,20 @@ const createStore = ({
         triggerEdgeChanges(edgeChanges);
       },
       setNodeExtent: (nodeExtent) => {
+        const { nodes, nodeLookup, parentLookup, nodeOrigin, elevateNodesOnSelect } = get();
+
         set({
           nodeExtent,
         });
-        get().setNodes(get().nodes);
+
+        adoptUserNodes(nodes, nodeLookup, parentLookup, {
+          nodeOrigin,
+          nodeExtent,
+          elevateNodesOnSelect,
+          checkEquality: false,
+        });
+
+        set({ nodes });
       },
       panBy: (delta): Promise<boolean> => {
         const { transform, width, height, panZoom, translateExtent } = get();