diff --git a/src/backend/langflow/interface/custom/custom_component.py b/src/backend/langflow/interface/custom/custom_component.py
index ce8956660..0d93f8d75 100644
--- a/src/backend/langflow/interface/custom/custom_component.py
+++ b/src/backend/langflow/interface/custom/custom_component.py
@@ -6,8 +6,8 @@ from langflow.interface.custom.directory_reader import DirectoryReader
 
 from langflow.utils import validate
 
-from langflow.database.base import session_getter
-from langflow.database.models.flow import Flow
+from langflow.services.database.base import session_getter
+from langflow.services.database.models.flow import Flow
 from pydantic import Extra
 import yaml
 
diff --git a/tests/test_cache_manager.py b/tests/test_cache_manager.py
index f3e65481e..660512634 100644
--- a/tests/test_cache_manager.py
+++ b/tests/test_cache_manager.py
@@ -2,7 +2,7 @@ from io import StringIO
 
 import pandas as pd
 import pytest
-from langflow.cache.manager import CacheManager
+from langflow.services.cache.manager import CacheManager
 from PIL import Image
 
 
diff --git a/tests/test_custom_component.py b/tests/test_custom_component.py
index 199906dda..f20311cec 100644
--- a/tests/test_custom_component.py
+++ b/tests/test_custom_component.py
@@ -5,7 +5,7 @@ from uuid import uuid4
 
 
 from fastapi import HTTPException
-from langflow.database.models.flow import Flow, FlowCreate
+from langflow.services.database.models.flow import Flow, FlowCreate
 from langflow.interface.custom.base import CustomComponent
 from langflow.interface.custom.component import (
     Component,
diff --git a/tests/test_database.py b/tests/test_database.py
index bc512b6b0..6ebae5396 100644
--- a/tests/test_database.py
+++ b/tests/test_database.py
@@ -8,9 +8,9 @@ from fastapi.testclient import TestClient
 from fastapi.encoders import jsonable_encoder
 
 from langflow.api.v1.schemas import FlowListCreate
-from langflow.database.models.flow import Flow, FlowCreate, FlowUpdate
+from langflow.services.database.models.flow import Flow, FlowCreate, FlowUpdate
 
-from langflow.database.models.flow_style import (
+from langflow.services.database.models.flow_style import (
     FlowStyleCreate,
     FlowStyleRead,
     FlowStyleUpdate,
diff --git a/tests/test_websocket.py b/tests/test_websocket.py
index 57a0e95f6..dd668c287 100644
--- a/tests/test_websocket.py
+++ b/tests/test_websocket.py
@@ -1,6 +1,6 @@
 from fastapi import WebSocketDisconnect
 
-# from langflow.chat.manager import ChatManager
+# from langflow.services.chat.manager import ChatManager
 
 import pytest