diff --git a/src/agents/tools/cron-tool.test-harness.ts b/src/agents/tools/cron-tool.test-harness.ts
index cf5c84564d..af8153f3b4 100644
--- a/src/agents/tools/cron-tool.test-harness.ts
+++ b/src/agents/tools/cron-tool.test-harness.ts
@@ -1,6 +1,7 @@
 import { vi } from "vitest";
+import type { MockFn } from "../../test-utils/vitest-mock-fn.js";
 
-export const callGatewayMock = vi.fn();
+export const callGatewayMock = vi.fn() as unknown as MockFn;
 
 vi.mock("../../gateway/call.js", () => ({
   callGateway: (opts: unknown) => callGatewayMock(opts),
diff --git a/src/commands/channels.mock-harness.ts b/src/commands/channels.mock-harness.ts
index e26703f60d..a71ae75d44 100644
--- a/src/commands/channels.mock-harness.ts
+++ b/src/commands/channels.mock-harness.ts
@@ -1,12 +1,18 @@
 import { vi } from "vitest";
+import type { MockFn } from "../test-utils/vitest-mock-fn.js";
 
-export const configMocks = {
-  readConfigFileSnapshot: vi.fn(),
-  writeConfigFile: vi.fn().mockResolvedValue(undefined),
+export const configMocks: {
+  readConfigFileSnapshot: MockFn;
+  writeConfigFile: MockFn;
+} = {
+  readConfigFileSnapshot: vi.fn() as unknown as MockFn,
+  writeConfigFile: vi.fn().mockResolvedValue(undefined) as unknown as MockFn,
 };
 
-export const offsetMocks = {
-  deleteTelegramUpdateOffset: vi.fn().mockResolvedValue(undefined),
+export const offsetMocks: {
+  deleteTelegramUpdateOffset: MockFn;
+} = {
+  deleteTelegramUpdateOffset: vi.fn().mockResolvedValue(undefined) as unknown as MockFn,
 };
 
 vi.mock("../config/config.js", async (importOriginal) => {