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