diff --git a/be/alembic/versions/0001_initial_schema.py b/be/alembic/versions/0001_initial_schema.py
index ad2aa7f..bf86155 100644
--- a/be/alembic/versions/0001_initial_schema.py
+++ b/be/alembic/versions/0001_initial_schema.py
@@ -25,7 +25,7 @@ recurrence_type_enum = postgresql.ENUM('DAILY', 'WEEKLY', 'MONTHLY', 'YEARLY', n
 chore_frequency_enum = postgresql.ENUM('one_time', 'daily', 'weekly', 'monthly', 'custom', name='chorefrequencyenum', create_type=False)
 chore_type_enum = postgresql.ENUM('personal', 'group', name='choretypeenum', create_type=False)
 
-def upgrade() -> None:
+def upgrade(context) -> None:
     user_role_enum.create(op.get_bind(), checkfirst=True)
     split_type_enum.create(op.get_bind(), checkfirst=True)
     expense_split_status_enum.create(op.get_bind(), checkfirst=True)