diff --git a/modules/tools/virsorter2.nf b/modules/tools/virsorter2.nf
index fc5d486..e891f21 100644
--- a/modules/tools/virsorter2.nf
+++ b/modules/tools/virsorter2.nf
@@ -3,16 +3,20 @@ process virsorter2 {
     errorStrategy 'ignore'
     input:
       tuple val(name), path(fasta)  
+      path(database)
     output:
         tuple val(name), path("virsorter2_*.out/final-viral-score.tsv")
         tuple val(name), path("virsorter2_*.out")
     script:
         """
-        virsorter run -w virsorter2_\${PWD##*/}.out -i ${fasta}  -j ${task.cpus}
+        virsorter run -d ${database} \
+        -w virsorter2_\${PWD##*/}.out \
+        -i ${fasta} \
+        -j ${task.cpus}
         """
 }
 
 
 //        tar cf virsorter2_results_\${PWD##*/}.out.tar virsorter2_\${PWD##*/}.out
 
-//
\ No newline at end of file
+//