diff --git a/hercules_opt/src/fork_transforms.rs b/hercules_opt/src/fork_transforms.rs
index ed6283fd2df09d096c40f38141060a046b4e7e80..c4a6ba7f23d0671e408639f9b7af6b1f73015dce 100644
--- a/hercules_opt/src/fork_transforms.rs
+++ b/hercules_opt/src/fork_transforms.rs
@@ -868,7 +868,7 @@ pub fn fork_dim_merge(
             };
             if tid_dim > inner_idx {
                 let new_tid = Node::ThreadID {
-                    control: new_fork_id,
+                    control: new_fork,
                     dimension: tid_dim - 1,
                 };
                 let new_tid = edit.add_node(new_tid);
@@ -876,7 +876,7 @@ pub fn fork_dim_merge(
                 edit.sub_edit(tid, new_tid);
             } else if tid_dim == outer_idx {
                 let outer_tid = Node::ThreadID {
-                    control: new_fork_id,
+                    control: new_fork,
                     dimension: outer_idx,
                 };
                 let outer_tid = edit.add_node(outer_tid);
@@ -894,7 +894,7 @@ pub fn fork_dim_merge(
                 edit = edit.replace_all_uses(tid, rem)?;
             } else if tid_dim == inner_idx {
                 let outer_tid = Node::ThreadID {
-                    control: new_fork_id,
+                    control: new_fork,
                     dimension: outer_idx,
                 };
                 let outer_tid = edit.add_node(outer_tid);