diff --git a/hercules_cg/src/gpu.rs b/hercules_cg/src/gpu.rs
index 3e0ae81dac796a85361a4df88067792655df581f..e7195223010777a79307d4dd38ef6c7d27272f16 100644
--- a/hercules_cg/src/gpu.rs
+++ b/hercules_cg/src/gpu.rs
@@ -1360,11 +1360,12 @@ extern \"C\" {} {}(", if ret_primitive { ret_type.clone() } else { "void".to_str
                 let collect_with_indices = self.codegen_collect(*collect, indices, is_char, extra_dim_collects.contains(&self.typing[collect.idx()]));
                 let data_variable = self.get_value(*data, false, false);
                 let data_type_id = self.typing[data.idx()];
-                let nested_fork = nesting_fork.unwrap();
                 let cg_tile = match state {
                     KernelState::OutBlock => "grid".to_string(),
                     KernelState::InBlock => "block".to_string(),
-                    KernelState::InThread => self.get_cg_tile(nested_fork, CGType::UsePerId),
+                    KernelState::InThread => {
+                        self.get_cg_tile(nesting_fork.unwrap(), CGType::UsePerId)
+                    }
                 };
                 if self.types[data_type_id.idx()].is_primitive() {
                     write!(w, "{}if ({}.thread_rank() == 0) {{\n", tabs, cg_tile)?;