diff --git a/juno_scheduler/src/pm.rs b/juno_scheduler/src/pm.rs
index 80699feebddb3d0ebacc8e9e9003330e78f1796b..b2845913a20467a5cdce1dfe11c83b95a213f492 100644
--- a/juno_scheduler/src/pm.rs
+++ b/juno_scheduler/src/pm.rs
@@ -1561,12 +1561,9 @@ fn run_pass(
                     let Some(mut func) = func else {
                         continue;
                     };
-                    // TODO: uses direct return from forkify for now instead of
-                    // func.modified, see comment on top of `forkify` for why. Fix
-                    // this eventually.
-                    let c = forkify(&mut func, control_subgraph, fork_join_map, loop_nest);
-                    changed |= c;
-                    inner_changed |= c;
+                    forkify(&mut func, control_subgraph, fork_join_map, loop_nest);
+                    changed |= func.modified();
+                    inner_changed |= func.modified();
                 }
                 pm.delete_gravestones();
                 pm.clear_analyses();