diff --git a/src/arriba_fusion.nf b/src/arriba_fusion.nf
index 53396918b74097b98df5d693a5d04db8ec5b719c..f657097ca82ac8bd7a65607eddb706a1dad94012 100644
--- a/src/arriba_fusion.nf
+++ b/src/arriba_fusion.nf
@@ -185,12 +185,12 @@ workflow {
     }
     filter_bam_quality(mapping2fusion.out.bam)
     index_bam(filter_bam_quality.out.bam.collect())
-    arriba(filter_bam_quality.out.bam, gtf_file.collect(), genome_file.collect())
-    draw_fusions(arriba.out.fusions, filter_bam_quality.out.bam, gtf_file, index_bam.out.bam_idx)
+    arriba(index_bam.out.bam_idx.collect(), gtf_file.collect(), genome_file.collect())
+    draw_fusions(arriba.out.fusions, index_bam.out.bam_idx, gtf_file)
   }
   else {
     index_bam(bam_files.collect())
-    arriba(bam_files, gtf_file.collect(), genome_file.collect())
-    draw_fusions(arriba.out.fusions, bam_files, gtf_file, index_bam.out.bam_idx)
+    arriba(index_bam.out.bam_idx.collect(), gtf_file.collect(), genome_file.collect())
+    draw_fusions(arriba.out.fusions, index_bam.out.bam_idx, gtf_file)
   }
 }
\ No newline at end of file
diff --git a/src/nf_modules/arriba/main.nf b/src/nf_modules/arriba/main.nf
index 0636d5b48c6510594288b4293e28311b63e26234..be1630ae081d7995c2479f7f2d0a307cb1db59a5 100644
--- a/src/nf_modules/arriba/main.nf
+++ b/src/nf_modules/arriba/main.nf
@@ -41,9 +41,8 @@ process draw_fusions{
 
   input:
   tuple val(fusion_id), path(fusions)
-  tuple val(bam_id), path(bam)
+  tuple val(bam_id), path(bam), path("*.bam.bai")
   path(gtf)
-  path(bai)
 
   output:
   tuple val(fusion_id), path("*.pdf"), emit: drawn_fusions