From 403bf71aeb85842f54a42b40712761fb39b3918c Mon Sep 17 00:00:00 2001
From: moklick <info@moritzklack.com>
Date: Sat, 28 May 2022 22:05:08 +0200
Subject: [PATCH] fix(uncontrolled-flow): keep internals

---
 src/hooks/useDrag/index.ts | 2 +-
 src/store/utils.ts         | 1 +
 src/types/changes.ts       | 3 +--
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/hooks/useDrag/index.ts b/src/hooks/useDrag/index.ts
index 67a4295e..8f0d6011 100644
--- a/src/hooks/useDrag/index.ts
+++ b/src/hooks/useDrag/index.ts
@@ -118,7 +118,7 @@ function useDrag({
             }
 
             event.on('end', (event) => {
-              setDragging(false)
+              setDragging(false);
               if (dragItems.current) {
                 updateNodePositions(dragItems.current, false, false);
 
diff --git a/src/store/utils.ts b/src/store/utils.ts
index 3bc89963..692dc7ca 100644
--- a/src/store/utils.ts
+++ b/src/store/utils.ts
@@ -143,6 +143,7 @@ export function handleControlledNodeSelectionChange(nodeChanges: NodeSelectionCh
     if (node) {
       nodeInternals.set(node.id, {
         ...node,
+        [internalsSymbol]: node[internalsSymbol],
         selected: change.selected,
       });
     }
diff --git a/src/types/changes.ts b/src/types/changes.ts
index d896d03e..b5eddaf6 100644
--- a/src/types/changes.ts
+++ b/src/types/changes.ts
@@ -1,12 +1,11 @@
 import { XYPosition, Dimensions } from './utils';
-import { NodeHandleBounds, Node } from './nodes';
+import { Node } from './nodes';
 import { Edge } from './edges';
 
 export type NodeDimensionChange = {
   id: string;
   type: 'dimensions';
   dimensions: Dimensions;
-  handleBounds?: NodeHandleBounds;
 };
 
 export type NodePositionChange = {