diff --git a/src/backend/langflow/api/v1/login.py b/src/backend/langflow/api/v1/login.py
index a11167a40..fc6d0444d 100644
--- a/src/backend/langflow/api/v1/login.py
+++ b/src/backend/langflow/api/v1/login.py
@@ -3,7 +3,7 @@ from fastapi import APIRouter, Depends, HTTPException, status
 from fastapi.security import OAuth2PasswordRequestForm
 
 from langflow.services.utils import get_session
-from langflow.services.database.models import Token
+from langflow.api.v1.schemas import Token
 from langflow.services.auth.utils import (
     authenticate_user,
     create_user_tokens,