diff --git a/flint/prefect/flows/continuum_pipeline.py b/flint/prefect/flows/continuum_pipeline.py index 73936089..ab1c1fd1 100644 --- a/flint/prefect/flows/continuum_pipeline.py +++ b/flint/prefect/flows/continuum_pipeline.py @@ -485,7 +485,8 @@ def process_science_fields( # zip up the final measurement set, which is not included in the above loop if field_options.zip_ms: - task_zip_ms.map(in_item=wsclean_cmds, wait_for=archive_wait_for) + zip_futures = task_zip_ms.map(in_item=wsclean_cmds, wait_for=archive_wait_for) + archive_wait_for.extend(zip_futures) if field_options.sbid_archive_path or field_options.sbid_copy_path: update_archive_options = get_options_from_strategy(