diff --git a/src/browser/pw-session.mock-setup.ts b/src/browser/pw-session.mock-setup.ts
index e62d51c9d14..0b176d536db 100644
--- a/src/browser/pw-session.mock-setup.ts
+++ b/src/browser/pw-session.mock-setup.ts
@@ -1,7 +1,8 @@
 import { vi } from "vitest";
+import type { MockFn } from "../test-utils/vitest-mock-fn.js";
 
-export const connectOverCdpMock = vi.fn();
-export const getChromeWebSocketUrlMock = vi.fn();
+export const connectOverCdpMock: MockFn = vi.fn();
+export const getChromeWebSocketUrlMock: MockFn = vi.fn();
 
 vi.mock("playwright-core", () => ({
   chromium: {