diff --git a/apps/studio/lib/ai/model.test.ts b/apps/studio/lib/ai/model.test.ts
index 7b5c58a4ae3..ffcf7055cd4 100644
--- a/apps/studio/lib/ai/model.test.ts
+++ b/apps/studio/lib/ai/model.test.ts
@@ -4,7 +4,7 @@ import * as bedrockModule from './bedrock'
 import { getModel, ModelErrorMessage } from './model'
 
 vi.mock('@ai-sdk/openai', () => ({
-  openai: { chat: vi.fn(() => 'openai-model') },
+  openai: vi.fn(() => 'openai-model'),
 }))
 
 vi.mock('./bedrock', async () => ({