diff --git a/cinn/hlir/pass/fusion_merge_pass.cc b/cinn/hlir/pass/fusion_merge_pass.cc
index 62b21db2b2..4243f04944 100644
--- a/cinn/hlir/pass/fusion_merge_pass.cc
+++ b/cinn/hlir/pass/fusion_merge_pass.cc
@@ -838,7 +838,7 @@ class FusionMergePassHelper : public FusionHelperBase {
           } else {
             for (auto& belong_group : cur->belong_groups) {
               if (belong_group->group_id == cur->group_id) {
-                updated_consumers.insert(cur);
+                updated_consumers.insert(belong_group);
               } else {
                 fused_groups.push(belong_group);
               }