diff --git a/src/scheduler.h b/src/scheduler.h
index b25ba843b29496439337c4876eb1b87f72367794..9bdb9b27da4089723dfcd6e93912f958f4811ab6 100644
--- a/src/scheduler.h
+++ b/src/scheduler.h
@@ -92,8 +92,8 @@ struct scheduler {
 };
 
 /* Function prototypes. */
-void scheduler_init(struct scheduler *s, struct space *space, int nr_queues,
-                    unsigned int flags, int nodeID);
+void scheduler_init(struct scheduler *s, struct space *space, int nr_tasks,
+                    int nr_queues, unsigned int flags, int nodeID);
 struct task *scheduler_gettask(struct scheduler *s, int qid,
                                struct cell *super);
 void scheduler_enqueue(struct scheduler *s, struct task *t);