diff --git a/workflowhub/generator/workflow/montage_recipe.py b/workflowhub/generator/workflow/montage_recipe.py index f0cdad08..06c462a5 100644 --- a/workflowhub/generator/workflow/montage_recipe.py +++ b/workflowhub/generator/workflow/montage_recipe.py @@ -289,6 +289,7 @@ def build_workflow(self, workflow_name: str = None) -> Workflow: input_files = [] for task in mbg_tasks: input_files.extend(self._get_files_by_task_and_link(task.name, FileLink.OUTPUT)) + input_files.extend(self._get_files_by_task_and_link(mimg_task.name, FileLink.OUTPUT)) madd_task = self._generate_task('mAdd', task_name, input_files=input_files, files_recipe={FileLink.OUTPUT: {'.fits': 2}}) workflow.add_node(task_name, task=madd_task)