diff --git a/endpoints/login.post.py b/endpoints/login.post.py
index 3fa91dd..0c4c62a 100644
--- a/endpoints/login.post.py
+++ b/endpoints/login.post.py
@@ -1,21 +1,19 @@
 from fastapi import APIRouter, Depends, HTTPException, status
 from sqlalchemy.orm import Session
 from core.database import get_db
-from schemas.user import UserLogin
-from helpers.auth_helpers import authenticate_user
+from schemas.user import UserSchema, UserCreate
+from helpers.user_helpers import get_user_by_username
 
 router = APIRouter()
 
-@router.post("/login", status_code=status.HTTP_200_OK)
+@router.post("/login", status_code=status.HTTP_200_OK, response_model=UserSchema)
 async def login(
-    user_data: UserLogin,
+    user_data: UserCreate,
     db: Session = Depends(get_db)
 ):
-    user = authenticate_user(db, user_data)
+    user = get_user_by_username(db, username=user_data.username)
     if not user:
-        raise HTTPException(
-            status_code=status.HTTP_401_UNAUTHORIZED,
-            detail="Incorrect email or password",
-            headers={"WWW-Authenticate": "Bearer"},
-        )
+        raise HTTPException(status_code=404, detail="User not found")
+    if not user.verify_password(user_data.password):
+        raise HTTPException(status_code=401, detail="Invalid password")
     return user
\ No newline at end of file