diff --git a/src/engine.c b/src/engine.c
index 6c1b0a446ac4b56715f0d9a75333e2e27f54b19f..e5a113e5b3cc099873480300f0d7afe9b051a72b 100644
--- a/src/engine.c
+++ b/src/engine.c
@@ -2368,7 +2368,7 @@ void engine_init_particles(struct engine *e, int flag_entropy_ICs,
 #endif
 
   scheduler_write_dependencies(&e->sched, e->verbose, e->step);
-  scheduler_write_cell_dependencies(&e->sched, e->verbose, e->step);
+  scheduler_write_cell_dependencies(&e->sched, e->verbose, e->step,  e->sched.dependency_graph_cellID);
   if (e->nodeID == 0) scheduler_write_task_level(&e->sched, e->step);
 
   /* Zero the list of cells that have had their time-step updated */
@@ -2904,7 +2904,7 @@ int engine_step(struct engine *e) {
   if (e->sched.frequency_dependency != 0 &&
       e->step % e->sched.frequency_dependency == 0) {
     scheduler_write_dependencies(&e->sched, e->verbose, e->step);
-    scheduler_write_cell_dependencies(&e->sched, e->verbose, e->step);
+    scheduler_write_cell_dependencies(&e->sched, e->verbose, e->step, e->sched.dependency_graph_cellID);
   }
 
   /* Write the task levels */