diff --git a/runtime/datamate-python/app/module/annotation/interface/task.py b/runtime/datamate-python/app/module/annotation/interface/task.py
index ca2c7a70..efe8c6e7 100644
--- a/runtime/datamate-python/app/module/annotation/interface/task.py
+++ b/runtime/datamate-python/app/module/annotation/interface/task.py
@@ -351,6 +351,9 @@ async def replace_file_tags(
     """
     service = DatasetManagementService(db)
 
+    from sqlalchemy.future import select
+    from app.db.models import DatasetFiles
+
     result = await db.execute(
         select(DatasetFiles).where(DatasetFiles.id == file_id)
     )