diff --git a/extensions/model-extension/src/index.test.ts b/extensions/model-extension/src/index.test.ts
index 05598c30d..e514f8ce3 100644
--- a/extensions/model-extension/src/index.test.ts
+++ b/extensions/model-extension/src/index.test.ts
@@ -1,5 +1,4 @@
 import JanModelExtension from './index'
-import { Model } from '@janhq/core'
 
 let SETTINGS = []
 // @ts-ignore
@@ -72,7 +71,7 @@ describe('JanModelExtension', () => {
   })
 
   it('should update a model', async () => {
-    const model: Partial<Model> = { id: 'test-model' }
+    const model = { id: 'test-model' }
     const updatedModel = await extension.updateModel(model)
     expect(updatedModel).toEqual({})
     expect(mockCortexAPI.updateModel).toHaveBeenCalledWith(model)
diff --git a/web/hooks/useModels.test.ts b/web/hooks/useModels.test.ts
index 33c152672..b58102144 100644
--- a/web/hooks/useModels.test.ts
+++ b/web/hooks/useModels.test.ts
@@ -1,5 +1,4 @@
 // useModels.test.ts
-
 import { renderHook, act } from '@testing-library/react'
 import { events, ModelEvent, ModelManager } from '@janhq/core'
 import { extensionManager } from '@/extension'
@@ -28,7 +27,9 @@ describe('useModels', () => {
     ;(ModelManager.instance as jest.Mock).mockReturnValue({
       models: {
         values: () => ({
-          toArray: () => {},
+          toArray: () => ({
+            filter: () => models,
+          }),
         }),
       },
     })