diff --git a/tests/main.imputation.nf.test b/tests/main.imputation.nf.test
index c1bdfcd..6c85552 100644
--- a/tests/main.imputation.nf.test
+++ b/tests/main.imputation.nf.test
@@ -53,7 +53,7 @@ nextflow_pipeline {
                 assert variantCount == TOTAL_REFPANEL_CHR20_B37 + ONLY_IN_INPUT
             }
             
-            with(zip.extract("chr20.info.gz").vcf) {
+            with(zip.extract("chr20.info.vcf.gz").vcf) {
                 assert variantCount == TOTAL_REFPANEL_CHR20_B37 + ONLY_IN_INPUT
             }
 
@@ -110,7 +110,7 @@ nextflow_pipeline {
                 assert variantCount == 7735
             }
 
-            with(zip.extract("chr20.info.gz").vcf) {
+            with(zip.extract("chr20.info.vcf.gz").vcf) {
                 assert variantCount == TOTAL_REFPANEL_CHR20_B37 + ONLY_IN_INPUT
             }
 
@@ -183,7 +183,7 @@ nextflow_pipeline {
                 assert variantCount == TOTAL_REFPANEL_CHR20_B37 + ONLY_IN_INPUT
             }
 
-            with(zip.extract("chr20.info.gz").vcf) {
+            with(zip.extract("chr20.info.vcf.gz").vcf) {
                 assert variantCount == TOTAL_REFPANEL_CHR20_B37 + ONLY_IN_INPUT
             }
 
@@ -231,7 +231,7 @@ nextflow_pipeline {
                 assert variantCount == TOTAL_REFPANEL_CHR20_B37
             }
 
-            with(zip.extract("chr20.info.gz").vcf) {
+            with(zip.extract("chr20.info.vcf.gz").vcf) {
                 assert variantCount == TOTAL_REFPANEL_CHR20_B37
             }         
 
@@ -278,7 +278,7 @@ nextflow_pipeline {
                 assert variantCount == TOTAL_REFPANEL_CHR20_B37
             }
 
-            with(path("${outputDir}/chr20.info.gz").vcf) {
+            with(path("${outputDir}/chr20.info.vcf.gz").vcf) {
                 assert variantCount == TOTAL_REFPANEL_CHR20_B37
             }
 
@@ -327,7 +327,7 @@ nextflow_pipeline {
                 assert variantCount == TOTAL_REFPANEL_CHR20_B37
             }
 
-            with(zip.extract("chr20.info.gz").vcf) {
+            with(zip.extract("chr20.info.vcf.gz").vcf) {
                 assert variantCount == TOTAL_REFPANEL_CHR20_B37
             }
 
@@ -422,7 +422,7 @@ nextflow_pipeline {
                 assert variantCount < TOTAL_REFPANEL_CHR20_B37
             }
 
-            with(zip.extract("chr20.info.gz").vcf) {
+            with(zip.extract("chr20.info.vcf.gz").vcf) {
                 assert variantCount == file.variantCount
             }