diff --git a/workflow/main.nf b/workflow/main.nf
index 1103db0f58bc9ccff692063b3e29c49a847fa4bb..bdaaa92114b16bd823251cc640c7b9de4ea4f601 100644
--- a/workflow/main.nf
+++ b/workflow/main.nf
@@ -38,6 +38,7 @@ readsList = Channel
 pairedEnd = params.pairedEnd
 designFile = params.designFile
 genomeSize = params.genomeSize
+genome = params.genome
 chromSizes = params.chromSizes
 fasta = params.fasta
 cutoffRatio = params.cutoffRatio
diff --git a/workflow/scripts/annotate_peaks.R b/workflow/scripts/annotate_peaks.R
index 98bb8bdc846442e22ff354165c4580fd7777e25c..3629c4acf85e87440e17f2846e3beb56ba2c43ce 100644
--- a/workflow/scripts/annotate_peaks.R
+++ b/workflow/scripts/annotate_peaks.R
@@ -21,16 +21,16 @@ if (length(args) != 2) {
 }
 
 design_file <- args[1]
-genome <-args[2]
+genome_assembly <- args[2]
 
 # Load UCSC Known Genes
-if(genome=='GRCh37') {
+if(genome_assembly=='GRCh37') {
     txdb <- TxDb.Hsapiens.UCSC.hg19.knownGene
     annodb <- 'org.Hs.eg.db'
-} else if(genome=='GRCm38')  {
+} else if(genome_assembly=='GRCm38')  {
     txdb <- TxDb.Mmusculus.UCSC.mm10.knownGene
     annodb <- 'org.Mm.eg.db'
-} else if(genome=='GRCh38')  {
+} else if(genome_assembly=='GRCh38')  {
     txdb <- TxDb.Hsapiens.UCSC.hg38.knownGene
     annodb <- 'org.Hs.eg.db'
 }