diff --git a/juno_scheduler/src/ir.rs b/juno_scheduler/src/ir.rs
index 48833b3cab26c7e7216e58861b423599d8c8a7ef..79ea858fb93c4e895f648f69935184242b98ed86 100644
--- a/juno_scheduler/src/ir.rs
+++ b/juno_scheduler/src/ir.rs
@@ -42,8 +42,8 @@ impl Pass {
     pub fn num_args(&self) -> usize {
         match self {
             Pass::Xdot => 1,
-            Pass::ForkChunk => 3,
-            Pass::ForkFissionBufferize => 4,
+            Pass::ForkChunk => 4,
+            Pass::ForkFissionBufferize => 2,
             Pass::ForkInterchange => 2,
             _ => 0,
         }