diff --git a/frontend/src/components/OcrRetryDialog/__tests__/OcrRetryDialog.test.tsx b/frontend/src/components/OcrRetryDialog/__tests__/OcrRetryDialog.test.tsx
index 4b42191..f766433 100644
--- a/frontend/src/components/OcrRetryDialog/__tests__/OcrRetryDialog.test.tsx
+++ b/frontend/src/components/OcrRetryDialog/__tests__/OcrRetryDialog.test.tsx
@@ -9,6 +9,7 @@ import { ocrService } from '../../../services/api';
 vi.mock('../../../services/api', () => ({
   ocrService: {
     getAvailableLanguages: vi.fn(),
+    getHealthStatus: vi.fn(),
     retryWithLanguage: vi.fn(),
   },
 }));
@@ -32,7 +33,16 @@ vi.mock('../../OcrLanguageSelector', () => ({
   ),
 }));
 
-const mockOcrService = vi.mocked(ocrService);
+const mockOcrService = {
+  getAvailableLanguages: vi.fn(),
+  getHealthStatus: vi.fn(),
+  retryWithLanguage: vi.fn(),
+} as any;
+
+// Replace the mocked service
+(ocrService as any).getAvailableLanguages = mockOcrService.getAvailableLanguages;
+(ocrService as any).getHealthStatus = mockOcrService.getHealthStatus;
+(ocrService as any).retryWithLanguage = mockOcrService.retryWithLanguage;
 
 const theme = createTheme();