diff --git a/src/backend/langflow/alembic/env.py b/src/backend/langflow/alembic/env.py
index a3babba6d..310894431 100644
--- a/src/backend/langflow/alembic/env.py
+++ b/src/backend/langflow/alembic/env.py
@@ -5,7 +5,7 @@ from sqlalchemy import pool
 
 from alembic import context
 
-from langflow.services.database.base import SQLModel
+from langflow.services.database.manager import SQLModel
 
 # this is the Alembic Config object, which provides
 # access to the values within the .ini file in use.
diff --git a/src/backend/langflow/interface/custom/custom_component.py b/src/backend/langflow/interface/custom/custom_component.py
index fdfef52f8..5388f40d8 100644
--- a/src/backend/langflow/interface/custom/custom_component.py
+++ b/src/backend/langflow/interface/custom/custom_component.py
@@ -7,7 +7,7 @@ from langflow.services.utils import get_db_manager
 
 from langflow.utils import validate
 
-from langflow.services.database.base import session_getter
+from langflow.services.database.utils import session_getter
 from langflow.services.database.models.flow import Flow
 from pydantic import Extra
 import yaml
diff --git a/src/backend/langflow/main.py b/src/backend/langflow/main.py
index 222873275..734483317 100644
--- a/src/backend/langflow/main.py
+++ b/src/backend/langflow/main.py
@@ -7,7 +7,7 @@ from fastapi.staticfiles import StaticFiles
 
 from langflow.api import router
 from langflow.interface.utils import setup_llm_caching
-from langflow.services.database.base import initialize_database
+from langflow.services.database.utils import initialize_database
 from langflow.services.manager import initialize_services
 from langflow.utils.logger import configure
 
diff --git a/src/backend/langflow/services/database/factory.py b/src/backend/langflow/services/database/factory.py
index d98414382..fecf24543 100644
--- a/src/backend/langflow/services/database/factory.py
+++ b/src/backend/langflow/services/database/factory.py
@@ -1,5 +1,5 @@
 from typing import TYPE_CHECKING
-from langflow.services.database.base import DatabaseManager
+from langflow.services.database.manager import DatabaseManager
 from langflow.services.factory import ServiceFactory
 
 if TYPE_CHECKING: