diff --git a/backend/app/alembic/env.py b/backend/app/alembic/env.py
index 7f29c04..b400d56 100755
--- a/backend/app/alembic/env.py
+++ b/backend/app/alembic/env.py
@@ -18,10 +18,10 @@ fileConfig(config.config_file_name)
 # target_metadata = mymodel.Base.metadata
 # target_metadata = None
 
-from app.models import SQLModel  # noqa
+from app.models.base import BaseSQLModel  # noqa
 from app.core.config import settings # noqa
 
-target_metadata = SQLModel.metadata
+target_metadata = BaseSQLModel.metadata
 
 # other values from the config, defined by the needs of env.py,
 # can be acquired: