diff --git a/tests/test_health.py b/tests/test_health.py
index 5b89cd8..2e92e97 100644
--- a/tests/test_health.py
+++ b/tests/test_health.py
@@ -1,6 +1,9 @@
-app = FastAPI()
+from fastapi import APIRouter, Depends, HTTPException
+from core.auth import get_current_user_dummy
+from core.database import fake_users_db
 
+router = APIRouter()
 
-@app.get("/health")
+@router.get("/health")
 async def read_health():
     return {"status": "ok"}
\ No newline at end of file