diff --git a/tests/wrappers/modules/icount_segment/test.yml b/tests/wrappers/modules/icount_segment/test.yml
index a3adfa72..73e74d88 100644
--- a/tests/wrappers/modules/icount_segment/test.yml
+++ b/tests/wrappers/modules/icount_segment/test.yml
@@ -1,5 +1,5 @@
 - name: "test_wrappers_icount_segment"
-  command: nextflow run ./wrappers/modules/icount_segment/main.nf -c ./tests/config/nextflow.config --gtf ./tests/data/genome/homosapien-hg37-chr21.fa.gz --fai ./tests/data/genome/homosapien-hg37-chr21.fa.fai
+  command: nextflow run ./wrappers/modules/icount_segment/main.nf -c ./tests/config/nextflow.config --gtf ./tests/data/genome/gencode.v35.chr21.gtf --fai ./tests/data/genome/homosapien-hg37-chr21.fa.fai
   tags:
     - "wrappers"
     - "wrappers/modules"
diff --git a/wrappers/modules/icount_segment/main.nf b/wrappers/modules/icount_segment/main.nf
index ddbbbfd4..26f8a220 100644
--- a/wrappers/modules/icount_segment/main.nf
+++ b/wrappers/modules/icount_segment/main.nf
@@ -7,6 +7,7 @@ include { ICOUNT_SEGMENT } from '../../../modules/goodwright/icount/segment/main
 workflow {
 
     ch_gtf = [ [id:file(params.gtf).baseName], file(params.gtf, checkIfExists: true) ]
+    ch_fai = file(params.fai, checkIfExists: true)
 
-    ICOUNT_SEGMENT ( ch_gtf, params.fai )
+    ICOUNT_SEGMENT ( ch_gtf, ch_fai )
 }