diff --git a/src/lib/storage.test.ts b/src/lib/storage.test.ts
index be78e56..2690ac8 100644
--- a/src/lib/storage.test.ts
+++ b/src/lib/storage.test.ts
@@ -2,10 +2,14 @@ import { describe, it, expect, vi, beforeEach } from 'vitest'
 
 const uploadMock = vi.fn()
 const getPublicUrlMock = vi.fn()
-const fromMock = vi.fn(() => ({ upload: uploadMock, getPublicUrl: getPublicUrlMock }))
+const fromMock = vi.fn()
 
-// @ts-expect-error - vitest mock function call compatibility
-vi.mock('./supabase', () => ({ supabase: { storage: { from: (...args: unknown[]) => fromMock(...args) } } }))
+vi.mock('./supabase', () => ({
+  supabase: { storage: { from: (bucketId: string) => {
+    fromMock(bucketId)
+    return { upload: uploadMock, getPublicUrl: getPublicUrlMock }
+  } } },
+}))
 
 import { uploadSalonImage } from './storage'