diff --git a/src/nf_modules/kb/main.nf b/src/nf_modules/kb/main.nf
index 5110b222d91721ed85d83f1240838b6cb5ce3b2b..8254a591fa2a059bccbf6ed28e9f7954e505053e 100644
--- a/src/nf_modules/kb/main.nf
+++ b/src/nf_modules/kb/main.nf
@@ -327,6 +327,7 @@ process velocity_default {
   if (reads.size() == 2)
   """
   mkdir ${file_prefix}
+  kb count  -t ${task.cpus} \
     -m ${kb_memory} \
     -i ${index} \
     -g ${transcript_to_gene} \