diff --git a/src/backend/langflow/alembic/env.py b/src/backend/langflow/alembic/env.py
index ea4fe9c43..a3babba6d 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.database.base import SQLModel
+from langflow.services.database.base import SQLModel
 
 # this is the Alembic Config object, which provides
 # access to the values within the .ini file in use.