diff --git a/video-gen-api/alembic/env.py b/video-gen-api/alembic/env.py
index c01fe41f..a18798e8 100644
--- a/video-gen-api/alembic/env.py
+++ b/video-gen-api/alembic/env.py
@@ -39,6 +39,7 @@ from app.models.resources_material import ResourcesMaterial  # noqa: F401
 from app.models.material_cost import MaterialCost  # noqa: F401
 from app.models.generated_resource import GeneratedResource  # noqa: F401
 from app.models.pre_test_template import PreTestTemplate  # noqa: F401
+from app.models.upload_task import UploadTask  # noqa: F401
 
 config = context.config