diff --git a/superset/migrations/versions/a9422eeaae74_new_dataset_models_take_2.py b/superset/migrations/versions/a9422eeaae74_new_dataset_models_take_2.py
index a7051f32c970d..9a2498bfa8590 100644
--- a/superset/migrations/versions/a9422eeaae74_new_dataset_models_take_2.py
+++ b/superset/migrations/versions/a9422eeaae74_new_dataset_models_take_2.py
@@ -42,7 +42,6 @@
 from sqlalchemy.sql.expression import and_, or_
 from sqlalchemy_utils import UUIDType
 
-from superset import app, db
 from superset.connectors.sqla.models import ADDITIVE_METRIC_TYPES_LOWER
 from superset.connectors.sqla.utils import get_dialect_name, get_identifier_quoter
 from superset.extensions import encrypted_field_factory
@@ -51,8 +50,6 @@
 from superset.utils.core import MediumText
 
 Base = declarative_base()
-custom_password_store = app.config["SQLALCHEMY_CUSTOM_PASSWORD_STORE"]
-DB_CONNECTION_MUTATOR = app.config["DB_CONNECTION_MUTATOR"]
 SHOW_PROGRESS = os.environ.get("SHOW_PROGRESS") == "1"
 UNKNOWN_TYPE = "UNKNOWN"
 
@@ -871,7 +868,7 @@ def reset_postgres_id_sequence(table: str) -> None:
 
 def upgrade() -> None:
     bind = op.get_bind()
-    session: Session = db.Session(bind=bind)
+    session: Session = Session(bind=bind)
     Base.metadata.drop_all(bind=bind, tables=new_tables)
     Base.metadata.create_all(bind=bind, tables=new_tables)