diff --git a/src/store/index.ts b/src/store/index.ts
index cb8b9502..bbb24cfe 100644
--- a/src/store/index.ts
+++ b/src/store/index.ts
@@ -233,6 +233,11 @@ const createStore = () =>
         nodeInternals: new Map(nodeInternals),
       });
     },
+    cancelConnection: () =>
+      set({
+        connectionNodeId: initialState.connectionNodeId,
+        connectionHandleId: initialState.connectionHandleId,
+      }),
     reset: () => set({ ...initialState }),
   }));