diff --git a/src/backend/langflow/services/database/manager.py b/src/backend/langflow/services/database/manager.py
index 60a4de74d..92385a457 100644
--- a/src/backend/langflow/services/database/manager.py
+++ b/src/backend/langflow/services/database/manager.py
@@ -4,7 +4,7 @@ from sqlmodel import SQLModel, Session, create_engine
 from langflow.utils.logger import logger
 from alembic.config import Config
 from alembic import command
-from .models import Flow  # noqa: F401
+from langflow.services.database import models  # noqa
 
 
 class DatabaseManager(Service):