diff --git a/be/alembic/migrations.py b/be/alembic/migrations.py
index ae3b924..1217b2d 100644
--- a/be/alembic/migrations.py
+++ b/be/alembic/migrations.py
@@ -23,8 +23,8 @@ def _get_migration_fn(script_directory, current_rev):
         # Get all upgrade steps from current revision to head
         revisions = script_directory._upgrade_revs("head", current_rev)
         for revision_step in revisions:
-            # Access the revision string from the RevisionStep object
-            script = script_directory.get_revision(revision_step.revision)
+            # Access the revision string from the Script object, which is within the RevisionStep object
+            script = script_directory.get_revision(revision_step.revision.revision)
             script.module.upgrade(context)
     return migration_fn