diff --git a/src/auto-reply/reply/dispatch-from-config.turn-model-selection-differential.test.ts b/src/auto-reply/reply/dispatch-from-config.turn-model-selection-differential.test.ts index f7539e444ace..012db2412af9 100644 --- a/src/auto-reply/reply/dispatch-from-config.turn-model-selection-differential.test.ts +++ b/src/auto-reply/reply/dispatch-from-config.turn-model-selection-differential.test.ts @@ -20,13 +20,9 @@ import { buildTestCtx } from "./test-ctx.js"; const selectAgentHarnessMock = vi.hoisted(() => vi.fn()); -vi.mock("../../agents/harness/selection.js", async (importOriginal) => { - const actual = await importOriginal(); - return { - ...actual, - selectAgentHarness: (...args: unknown[]) => selectAgentHarnessMock(...args), - }; -}); +vi.mock("../../agents/harness/selection.js", () => ({ + selectAgentHarness: (...args: unknown[]) => selectAgentHarnessMock(...args), +})); const { resolveVisibleRepliesPolicy } = await import("./dispatch-from-config.harness-defaults.js");