diff --git a/frontend/src/test/setup.ts b/frontend/src/test/setup.ts
index 6bd4095..f3d19aa 100644
--- a/frontend/src/test/setup.ts
+++ b/frontend/src/test/setup.ts
@@ -2,7 +2,7 @@
 // This file is automatically loaded before all tests
 
 import '@testing-library/jest-dom'
-import { vi } from 'vitest'
+import { vi, beforeEach, afterEach } from 'vitest'
 import { setupTestEnvironment } from './test-utils.tsx'
 
 // Setup global test environment