diff --git a/apps/desktop/src/lib/model-options.test.ts b/apps/desktop/src/lib/model-options.test.ts
index 7ad3e23a2e..f7fde1e338 100644
--- a/apps/desktop/src/lib/model-options.test.ts
+++ b/apps/desktop/src/lib/model-options.test.ts
@@ -132,7 +132,10 @@ describe('requestModelOptions', () => {
     const gateway = {
       request: vi.fn(() => Promise.resolve(gatewayPayload))
     }
-    const request = vi.fn(() => Promise.resolve(routedPayload))
+    const request = vi.fn(() => Promise.resolve(routedPayload)) as unknown as <T>(
+      method: string,
+      params?: Record<string, unknown>
+    ) => Promise<T>
 
     await expect(
       requestModelOptions({ gateway: gateway as never, request, sessionId: 'tile-1' })
diff --git a/apps/desktop/src/store/session-states-scopes.test.ts b/apps/desktop/src/store/session-states-scopes.test.ts
index 55169e3ba6..c8ab20a599 100644
--- a/apps/desktop/src/store/session-states-scopes.test.ts
+++ b/apps/desktop/src/store/session-states-scopes.test.ts
@@ -1,4 +1,4 @@
-import { beforeEach, describe, expect, it } from 'vitest'
+import { afterEach, beforeEach, describe, expect, it } from 'vitest'
 
 import { createClientSessionState } from '@/lib/chat-runtime'
 import {