diff --git a/src/main.py b/src/main.py
index 65a88f3..0d48b9b 100644
--- a/src/main.py
+++ b/src/main.py
@@ -110,5 +110,5 @@ def download(project: sly.Project) -> str:
     sly.output.set_download(download_dir)
     sly.logger.info("Archive uploaded and ready for download.")
     api.app.add_input_project(project)
-    file_id = api.file.get_info_by_path(download_dir).id
+    file_id = api.file.get_info_by_path(project.team_id, download_dir).id
     api.app.add_output_file(file_id)