diff --git a/src/components/__tests__/AudioInputNode.test.tsx b/src/components/__tests__/AudioInputNode.test.tsx
index 4b38d537..98af4e6d 100644
--- a/src/components/__tests__/AudioInputNode.test.tsx
+++ b/src/components/__tests__/AudioInputNode.test.tsx
@@ -39,6 +39,7 @@ vi.mock("@xyflow/react", () => {
       setNodes: () => {},
       screenToFlowPosition: (pos: unknown) => pos,
     }),
+    useConnection: (selector: (state: { inProgress: boolean }) => boolean) => selector({ inProgress: false }),
   };
 });
 
diff --git a/src/components/__tests__/EaseCurveNode.test.tsx b/src/components/__tests__/EaseCurveNode.test.tsx
index 5f209fac..b2ea320b 100644
--- a/src/components/__tests__/EaseCurveNode.test.tsx
+++ b/src/components/__tests__/EaseCurveNode.test.tsx
@@ -40,6 +40,7 @@ vi.mock("@xyflow/react", () => {
       setNodes: vi.fn(),
       screenToFlowPosition: (pos: unknown) => pos,
     }),
+    useConnection: (selector: (state: { inProgress: boolean }) => boolean) => selector({ inProgress: false }),
   };
 });
 
diff --git a/src/components/__tests__/VideoStitchNode.test.tsx b/src/components/__tests__/VideoStitchNode.test.tsx
index 657928c5..bfbc966e 100644
--- a/src/components/__tests__/VideoStitchNode.test.tsx
+++ b/src/components/__tests__/VideoStitchNode.test.tsx
@@ -42,6 +42,7 @@ vi.mock("@xyflow/react", () => {
       setNodes: () => {},
       screenToFlowPosition: (pos: unknown) => pos,
     }),
+    useConnection: (selector: (state: { inProgress: boolean }) => boolean) => selector({ inProgress: false }),
   };
 });