diff --git a/client/tests/setup.ts b/client/tests/setup.ts
index 0c0fc012..4455d6fe 100644
--- a/client/tests/setup.ts
+++ b/client/tests/setup.ts
@@ -11,6 +11,8 @@ vi.mock('../src/api/websocket', () => ({
   getSocketId: vi.fn(() => null),
   setRefetchCallback: vi.fn(),
   setPreReconnectHook: vi.fn(),
+  addListener: vi.fn(),
+  removeListener: vi.fn(),
 }));
 
 // MSW lifecycle