diff --git a/system/main.c b/system/main.c
index 0c50142cff..5d74cf20da 100644
--- a/system/main.c
+++ b/system/main.c
@@ -94,15 +94,15 @@ int main(int argc, char **argv)
     hwaddr end = (hwaddr)strtoll(argv[3], NULL, 16);
     input_size = atoi(argv[4]);
     char *output_path = argv[5];
+    unsigned int num_tasks = atoi(argv[6]);
     // hwaddr target_addr = (hwaddr) strtoll(argv[1], NULL, 16);
     // vm_start();
     // fix arguments for qemu
-    argv[5] = argv[0];
-    argv = &argv[5];
+    argv[6] = argv[0];
+    argv = &argv[6];
 
-    argc -= 5;
+    argc -= 6;
 
-    unsigned int num_tasks = 2;
     int full_input_room = (int)pow(input_size, num_tasks);
     printf("Full input room: %d\n", full_input_room);
     unsigned long *deltas = malloc(full_input_room * sizeof(unsigned long));