diff --git a/test/setupFlowTests.ts b/test/setupFlowTests.ts
index b625c03..99920df 100644
--- a/test/setupFlowTests.ts
+++ b/test/setupFlowTests.ts
@@ -63,7 +63,20 @@ export const mockReactFlow = () => {
   });
 };
 
-beforeEach(() => { useFlowStore.setState({ nodes: [], edges: [] }); });
+beforeAll(() => {
+  useFlowStore.setState({
+    nodes: [],
+    edges: [],
+    edgeReconnectSuccessful: true
+  });
+});
 
-afterEach(() => { cleanup(); });
+afterEach(() => {
+  cleanup();
+  useFlowStore.setState({
+    nodes: [],
+    edges: [],
+    edgeReconnectSuccessful: true
+  });
+});