diff --git a/src/backend/langflow/alembic/versions/eb5866d51fd2_change_columns_to_be_nullable.py b/src/backend/langflow/alembic/versions/eb5866d51fd2_change_columns_to_be_nullable.py
index e08641ffd..10d503e60 100644
--- a/src/backend/langflow/alembic/versions/eb5866d51fd2_change_columns_to_be_nullable.py
+++ b/src/backend/langflow/alembic/versions/eb5866d51fd2_change_columns_to_be_nullable.py
@@ -9,7 +9,7 @@ from typing import Sequence, Union
 
 from alembic import op
 import sqlalchemy as sa
-
+import sqlmodel  # noqa: F401
 
 # revision identifiers, used by Alembic.
 revision: str = "eb5866d51fd2"