diff --git a/tests/ut/ge/hybrid/ge_hybrid_unittest.cc b/tests/ut/ge/hybrid/ge_hybrid_unittest.cc
index 57230f30..18bcd7da 100644
--- a/tests/ut/ge/hybrid/ge_hybrid_unittest.cc
+++ b/tests/ut/ge/hybrid/ge_hybrid_unittest.cc
@@ -276,9 +276,9 @@ TEST_F(UtestGeHybrid, test_parse_parallel_group) {
   op_desc->SetOpKernelLibName("ops_kernel_info_hccl");
   GeRootModelPtr root_model = MakeShared<ge::GeRootModel>(compute_graph);
   HybridModel model(root_model);
+  model.root_graph_ = compute_graph;
 
   HybridModelBuilder builder(model);
-  builder.root_graph_ = compute_graph;
   ASSERT_EQ(builder.CollectParallelGroups(node_item.get()), SUCCESS);
 
   ASSERT_EQ(builder.node_to_parallel_groups_.size(), 1);