From 5954fa6cc4cdc0038d5c34010845bcb9288c3d55 Mon Sep 17 00:00:00 2001
From: peterkogo <peter@xyflow.com>
Date: Mon, 14 Oct 2024 11:44:44 +0200
Subject: [PATCH] remove non-null assertion while dragging

---
 packages/system/src/xydrag/utils.ts | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/packages/system/src/xydrag/utils.ts b/packages/system/src/xydrag/utils.ts
index a668746a..0fc048e8 100644
--- a/packages/system/src/xydrag/utils.ts
+++ b/packages/system/src/xydrag/utils.ts
@@ -106,14 +106,16 @@ export function getEventHandlerParams<NodeType extends InternalNodeBase>({
     return [nodesFromDragItems[0], nodesFromDragItems];
   }
 
-  const node = nodeLookup.get(nodeId)!.internals.userNode;
+  const node = nodeLookup.get(nodeId)?.internals.userNode;
 
   return [
-    {
-      ...node,
-      position: dragItems.get(nodeId)?.position || node.position,
-      dragging,
-    },
+    !node
+      ? nodesFromDragItems[0]
+      : {
+          ...node,
+          position: dragItems.get(nodeId)?.position || node.position,
+          dragging,
+        },
     nodesFromDragItems,
   ];
 }