diff --git a/packages/core/src/store/index.ts b/packages/core/src/store/index.ts
index 37d6ad2b..5310992a 100644
--- a/packages/core/src/store/index.ts
+++ b/packages/core/src/store/index.ts
@@ -238,6 +238,11 @@ const createRFStore = () =>
         nodeInternals: new Map(nodeInternals),
       });
     },
+    cancelConnection: () =>
+      set({
+        connectionNodeId: initialState.connectionNodeId,
+        connectionHandleId: initialState.connectionHandleId,
+      }),
     reset: () => set({ ...initialState }),
   }));