diff --git a/labelu/alembic_labelu/versions/bc8fcb35b66b_add_media_and_pre_annotation.py b/labelu/alembic_labelu/versions/bc8fcb35b66b_add_media_and_pre_annotation.py
index 94dc625..ade6b26 100644
--- a/labelu/alembic_labelu/versions/bc8fcb35b66b_add_media_and_pre_annotation.py
+++ b/labelu/alembic_labelu/versions/bc8fcb35b66b_add_media_and_pre_annotation.py
@@ -155,6 +155,7 @@ def upgrade() -> None:
                         f"UPDATE task_sample SET file_id={file_id} WHERE id={task_sample_id}"
                     )
         
+        session.commit()
     
 
 def downgrade() -> None: