diff --git a/superset/migrations/versions/abe27eaf93db_add_extra_config_column_to_alerts.py b/superset/migrations/versions/abe27eaf93db_add_extra_config_column_to_alerts.py
index c0cce0ac9e650..daff2be133564 100644
--- a/superset/migrations/versions/abe27eaf93db_add_extra_config_column_to_alerts.py
+++ b/superset/migrations/versions/abe27eaf93db_add_extra_config_column_to_alerts.py
@@ -31,12 +31,11 @@
 from sqlalchemy import String
 from sqlalchemy.sql import column, table
 
-connection = op.get_bind()
-
 report_schedule = table("report_schedule", column("extra", String))
 
 
 def upgrade():
+    connection = op.get_bind()
     with op.batch_alter_table("report_schedule") as batch_op:
         batch_op.add_column(
             sa.Column("extra", sa.Text(), nullable=True, default="{}",),