diff --git a/endpoints/login.post.py b/endpoints/login.post.py
index c0646d5..0b121d9 100644
--- a/endpoints/login.post.py
+++ b/endpoints/login.post.py
@@ -6,11 +6,11 @@ from sqlalchemy.orm import Session
 from typing import List
 from core.database import get_db
 from models.message import Message
-from schemas.message import MessageSchema
+from schemas.message import MessageSchema, MessageCreate
 
 router = APIRouter()
 
-@router.get("/messages", status_code=200, response_model=List[MessageSchema])
+@router.get("/messages/{user_id}", response_model=List[MessageSchema])
 async def get_user_messages(
     user_id: int,
     db: Session = Depends(get_db)
@@ -18,5 +18,12 @@ async def get_user_messages(
     messages = db.query(Message).filter(
         (Message.sender_id == user_id) | (Message.receiver_id == user_id)
     ).all()
+    
+    if not messages:
+        raise HTTPException(
+            status_code=status.HTTP_404_NOT_FOUND,
+            detail="No messages found for this user"
+        )
+    
     return messages
 ```
\ No newline at end of file