diff --git a/src/cron/isolated-agent/run.cron-model-override-forwarding.test.ts b/src/cron/isolated-agent/run.cron-model-override-forwarding.test.ts index 88cb170a1fb..d9429718333 100644 --- a/src/cron/isolated-agent/run.cron-model-override-forwarding.test.ts +++ b/src/cron/isolated-agent/run.cron-model-override-forwarding.test.ts @@ -177,11 +177,9 @@ describe("runCronIsolatedAgentTurn — cron model override forwarding (#58065)", const result = await runCronIsolatedAgentTurn(makeParams()); expect(result.status).toBe("ok"); - const embeddedCall = runEmbeddedPiAgentMock.mock.calls[0]?.[0] as - | { provider?: string; model?: string } - | undefined; - expect(embeddedCall?.provider).toBe("google"); - expect(embeddedCall?.model).toBe("gemini-2.0-flash"); + const embeddedCall = firstMockArg(runEmbeddedPiAgentMock); + expect(embeddedCall.provider).toBe("google"); + expect(embeddedCall.model).toBe("gemini-2.0-flash"); }); it("forwards isolated cron execution phase updates from embedded runs", async () => {