diff --git a/src/nf_modules/flexi_splitter/main.nf b/src/nf_modules/flexi_splitter/main.nf
index 6e01c268ee29352643956903869e8436f90f01ef..d7a87b94a28fdb48131c0edcdc7afcd4d47c9319 100644
--- a/src/nf_modules/flexi_splitter/main.nf
+++ b/src/nf_modules/flexi_splitter/main.nf
@@ -30,10 +30,10 @@ process split {
 
   if (reads.size() == 2)
   """
-  flexi_splitter ${params.split} -f ${reads[0]} ${reads[1]} -c ${config} -o split
+  flexi_splitter ${params.split} -n 2 -f ${reads[0]} ${reads[1]} -o split -c ${config}
   """
   else
   """
-  flexi_splitter ${params.split} -f ${reads[0]} -c ${config} -o split
+  flexi_splitter ${params.split} -n 1 -f ${reads[0]} -o split -c ${config}
   """
 }
diff --git a/src/nf_modules/kb/main.nf b/src/nf_modules/kb/main.nf
index fe08a0df8093a34ef4eb5026a85bee6d6db60ac1..0dcc470d39832bc394cad20c9b1aa5e2c2ac94fc 100644
--- a/src/nf_modules/kb/main.nf
+++ b/src/nf_modules/kb/main.nf
@@ -137,6 +137,7 @@ process kb_default {
   tuple val(file_id), path("*_report.txt"), emit: report
 
   script:
+  def kb_memory = "${task.memory}" - ~/GB/
   if (file_id instanceof List){
     file_prefix = file_id[0]
   } else {
@@ -151,7 +152,7 @@ process kb_default {
   """
   mkdir ${file_prefix}
   kb count  -t ${task.cpus} \
-    -m ${task.memory} \
+    -m ${kb_memory} \
     -i ${index} \
     -g ${transcript_to_gene} \
     ${whitelist_param} \
@@ -185,7 +186,7 @@ process kb_marseq {
   tuple val(file_id), path("*_report.txt"), emit: report
 
   script:
-  memory = "${task.memory}" - ~/\s*GB/
+  def kb_memory = "${task.memory}" - ~/GB/
   if (file_id instanceof List){
     file_prefix = file_id[0]
   } else {
@@ -200,7 +201,7 @@ process kb_marseq {
   """
   mkdir ${file_prefix}
   kb count  -t ${task.cpus} \
-    -m ${memory} \
+    -m ${kb_memory} \
     -i ${index} \
     -g ${transcript_to_gene} \
     ${whitelist_param} \