diff --git a/hercules_test/hercules_tests/tests/opt_tests.rs b/hercules_test/hercules_tests/tests/opt_tests.rs
index c14d4db5b0072ec0f724cdde9b467c7568e1b41c..256ab2ee3bc107e96f5511db932b10d9b16b885c 100644
--- a/hercules_test/hercules_tests/tests/opt_tests.rs
+++ b/hercules_test/hercules_tests/tests/opt_tests.rs
@@ -2,6 +2,7 @@ use std::env;
 
 use hercules_interpreter::*;
 use hercules_opt::pass::Pass;
+use hercules_ir::ID;
 
 extern crate rand;
 use rand::Rng;
diff --git a/hercules_test/test_inputs/matmul_int.hir b/hercules_test/test_inputs/matmul_int.hir
index 1e496babc9aaa8877f4607e2f8d572a0e5ce9e25..34d8169b5b721b527cc0497579cef76d0f29c0bd 100644
--- a/hercules_test/test_inputs/matmul_int.hir
+++ b/hercules_test/test_inputs/matmul_int.hir
@@ -1,11 +1,11 @@
 fn matmul<3>(a: array(i32, #0, #1), b: array(i32, #1, #2)) -> array(i32, #0, #2)
   c = constant(array(i32, #0, #2), [])
   i_ctrl = fork(start, #0)
-  i_idx = thread_id(i_ctrl)
+  i_idx = thread_id(i_ctrl, 0)
   j_ctrl = fork(i_ctrl, #2)
-  j_idx = thread_id(j_ctrl)
+  j_idx = thread_id(j_ctrl, 0)
   k_ctrl = fork(j_ctrl, #1)
-  k_idx = thread_id(k_ctrl)
+  k_idx = thread_id(k_ctrl, 0)
   k_join_ctrl = join(k_ctrl)
   j_join_ctrl = join(k_join_ctrl)
   i_join_ctrl = join(j_join_ctrl)