diff --git a/src/agents/model-auth-label.test.ts b/src/agents/model-auth-label.test.ts
index a7cb686a8c9..66fc0b4cdb5 100644
--- a/src/agents/model-auth-label.test.ts
+++ b/src/agents/model-auth-label.test.ts
@@ -6,7 +6,7 @@ const mocks = vi.hoisted(() => ({
   resolveAuthProfileOrder: vi.fn(),
   resolveAuthProfileDisplayLabel: vi.fn(),
   resolveUsableCustomProviderApiKey: vi.fn(() => null),
-  resolveEnvApiKey: vi.fn(() => null),
+  resolveEnvApiKey: vi.fn<() => { apiKey: string; source: string } | null>(() => null),
   readCodexCliCredentialsCached: vi.fn<() => unknown>(() => null),
 }));