From 87d363371b2942277b651bdec537ef1e2b55f3ff Mon Sep 17 00:00:00 2001
From: moklick <info@moritzklack.com>
Date: Thu, 29 Feb 2024 19:44:45 +0100
Subject: [PATCH] fix(react): handle drag for nodes that are initially hidden

---
 packages/react/src/hooks/useDrag.ts | 34 ++++++++++++++---------------
 1 file changed, 16 insertions(+), 18 deletions(-)

diff --git a/packages/react/src/hooks/useDrag.ts b/packages/react/src/hooks/useDrag.ts
index 52d3c6f6..ee26cc57 100644
--- a/packages/react/src/hooks/useDrag.ts
+++ b/packages/react/src/hooks/useDrag.ts
@@ -31,24 +31,22 @@ export function useDrag({
   const xyDrag = useRef<XYDragInstance>();
 
   useEffect(() => {
-    if (nodeRef?.current) {
-      xyDrag.current = XYDrag({
-        getStoreItems: () => store.getState(),
-        onNodeMouseDown: (id: string) => {
-          handleNodeClick({
-            id,
-            store,
-            nodeRef,
-          });
-        },
-        onDragStart: () => {
-          setDragging(true);
-        },
-        onDragStop: () => {
-          setDragging(false);
-        },
-      });
-    }
+    xyDrag.current = XYDrag({
+      getStoreItems: () => store.getState(),
+      onNodeMouseDown: (id: string) => {
+        handleNodeClick({
+          id,
+          store,
+          nodeRef,
+        });
+      },
+      onDragStart: () => {
+        setDragging(true);
+      },
+      onDragStop: () => {
+        setDragging(false);
+      },
+    });
   }, []);
 
   useEffect(() => {