diff --git a/src/renderer/src/aiCore/legacy/clients/__tests__/index.clientCompatibilityTypes.test.ts b/src/renderer/src/aiCore/legacy/clients/__tests__/index.clientCompatibilityTypes.test.ts
index 3157e4b63..ba7706522 100644
--- a/src/renderer/src/aiCore/legacy/clients/__tests__/index.clientCompatibilityTypes.test.ts
+++ b/src/renderer/src/aiCore/legacy/clients/__tests__/index.clientCompatibilityTypes.test.ts
@@ -10,12 +10,12 @@ import type { EndpointType, Model, Provider } from '@renderer/types'
 import { beforeEach, describe, expect, it, vi } from 'vitest'
 
 vi.mock('@renderer/config/models', () => ({
+  DEFAULT_MODEL_MAP: {
+    assistant: { id: 'gpt-4', name: 'GPT-4' },
+    quick: { id: 'gpt-4', name: 'GPT-4' },
+    translate: { id: 'gpt-4', name: 'GPT-4' }
+  },
   SYSTEM_MODELS: {
-    defaultModel: [
-      { id: 'gpt-4', name: 'GPT-4' },
-      { id: 'gpt-4', name: 'GPT-4' },
-      { id: 'gpt-4', name: 'GPT-4' }
-    ],
     zhipu: [],
     silicon: [],
     openai: [],