diff --git a/endpoints/logout.post.py b/endpoints/logout.post.py
index 91b2c5b..1b0f199 100644
--- a/endpoints/logout.post.py
+++ b/endpoints/logout.post.py
@@ -1,17 +1,11 @@
 # Entity: User
 
 from fastapi import APIRouter, Depends, status
-from sqlalchemy.orm import Session
-from core.database import get_db
-from models.user import User
-from schemas.user import UserSchema
-from helpers.user_helpers import logout_user
+from core.auth import logout_user
+from schemas.user import UserLogoutSchema
 
 router = APIRouter()
 
-@router.post("/logout", status_code=200)
-async def logout(
-    db: Session = Depends(get_db)
-):
-    logout_user(db)
+@router.post("/logout", status_code=200, response_model=UserLogoutSchema)
+async def logout(user=Depends(logout_user)):
     return {"message": "Logged out successfully"}
\ No newline at end of file