diff --git a/src/web/test-helpers.ts b/src/web/test-helpers.ts
index 78abd86a98..c3d2312777 100644
--- a/src/web/test-helpers.ts
+++ b/src/web/test-helpers.ts
@@ -65,22 +65,20 @@ vi.mock("qrcode-terminal", () => ({
   generate: vi.fn(),
 }));
 
-export const baileys =
-  (await import("@whiskeysockets/baileys")) as unknown as typeof import("@whiskeysockets/baileys") & {
-    makeWASocket: ReturnType<typeof vi.fn>;
-    useMultiFileAuthState: ReturnType<typeof vi.fn>;
-    fetchLatestBaileysVersion: ReturnType<typeof vi.fn>;
-    makeCacheableSignalKeyStore: ReturnType<typeof vi.fn>;
-  };
+export const baileys = await import("@whiskeysockets/baileys");
 
 export function resetBaileysMocks() {
   const recreated = createMockBaileys();
   (globalThis as Record<PropertyKey, unknown>)[Symbol.for("openclaw:lastSocket")] =
     recreated.lastSocket;
-  baileys.makeWASocket.mockImplementation(recreated.mod.makeWASocket);
-  baileys.useMultiFileAuthState.mockImplementation(recreated.mod.useMultiFileAuthState);
-  baileys.fetchLatestBaileysVersion.mockImplementation(recreated.mod.fetchLatestBaileysVersion);
-  baileys.makeCacheableSignalKeyStore.mockImplementation(recreated.mod.makeCacheableSignalKeyStore);
+  // @ts-expect-error
+  baileys.makeWASocket = vi.fn(recreated.mod.makeWASocket);
+  // @ts-expect-error
+  baileys.useMultiFileAuthState = vi.fn(recreated.mod.useMultiFileAuthState);
+  // @ts-expect-error
+  baileys.fetchLatestBaileysVersion = vi.fn(recreated.mod.fetchLatestBaileysVersion);
+  // @ts-expect-error
+  baileys.makeCacheableSignalKeyStore = vi.fn(recreated.mod.makeCacheableSignalKeyStore);
 }
 
 export function getLastSocket(): MockBaileysSocket {
diff --git a/tsconfig.json b/tsconfig.json
index 4e6d616c38..2235157723 100644
--- a/tsconfig.json
+++ b/tsconfig.json
@@ -20,11 +20,5 @@
     "useDefineForClassFields": false
   },
   "include": ["src/**/*", "ui/**/*"],
-  "exclude": [
-    "node_modules",
-    "dist",
-    "src/**/*.test.ts",
-    "src/**/*.test.tsx",
-    "src/**/test-helpers.ts"
-  ]
+  "exclude": ["node_modules", "dist", "src/**/*.test.ts"]
 }