diff --git a/endpoints/login.post.py b/endpoints/login.post.py
index 908eeb8..91bea63 100644
--- a/endpoints/login.post.py
+++ b/endpoints/login.post.py
@@ -5,8 +5,8 @@ from fastapi import APIRouter, Depends, HTTPException, status
 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 core.models.message import Message
+from core.schemas.message import MessageSchema, MessageCreate
 
 router = APIRouter()