diff --git a/extensions/openai/image-generation-provider.test.ts b/extensions/openai/image-generation-provider.test.ts
index 39ac33fded3..15d7b4eddf7 100644
--- a/extensions/openai/image-generation-provider.test.ts
+++ b/extensions/openai/image-generation-provider.test.ts
@@ -9,7 +9,9 @@ const {
   assertOkOrThrowHttpErrorMock,
   resolveProviderHttpRequestConfigMock,
 } = vi.hoisted(() => ({
-  isProviderApiKeyConfiguredMock: vi.fn(() => false),
+  isProviderApiKeyConfiguredMock: vi.fn<
+    (params: { provider: string; agentDir?: string }) => boolean
+  >(() => false),
   resolveApiKeyForProviderMock: vi.fn(
     async (_params?: {
       provider?: string;