diff --git a/endpoints/logout.post.py b/endpoints/logout.post.py
index dcad03c..5aab278 100644
--- a/endpoints/logout.post.py
+++ b/endpoints/logout.post.py
@@ -9,16 +9,15 @@ async def logout_handler(
     db: Session = Depends(get_db)
 ):
     """Demo logout endpoint"""
-    user = get_user_from_token(token, db)
+    user = get_user_from_token(token, fake_users_db)
     if not user:
         raise HTTPException(status_code=401, detail="Invalid authentication credentials")
     
-    # Invalidate token or remove user session
-    invalidate_token(token)
+    # Invalidate token or update user session
+    revoke_token(token)
+    user["active_session"] = None
     
     return {
         "message": "Logout successful",
-        "next_steps": [
-            "Redirect to login page"
-        ]
+        "user_id": user["id"]
     }
\ No newline at end of file