diff --git a/tests/ut/ge/hybrid/ge_hybrid_unittest.cc b/tests/ut/ge/hybrid/ge_hybrid_unittest.cc
index e8e8e196..de84342c 100644
--- a/tests/ut/ge/hybrid/ge_hybrid_unittest.cc
+++ b/tests/ut/ge/hybrid/ge_hybrid_unittest.cc
@@ -264,7 +264,7 @@ TEST_F(UtestGeHybrid, unfold_subgraphs_success) {
   OpDescPtr sub_sub_graph_while_cond_data_op_desc = CreateOpDesc("cond_data", DATA);
   NodePtr sub_sub_graph_while_cond_data_node = sub_sub_graph1->AddNode(sub_sub_graph_while_cond_data_op_desc);
 
-  ComputeGraphPtr sub_sub_graph2 = std::make_shared<ComputeGraph>("while body");
+  ComputeGraphPtr sub_sub_graph2 = std::make_shared<ComputeGraph>("while_body");
   /*OpDescPtr sub_sub_graph_while_body_const_op_desc = CreateOpDesc("body_const", CONSTANT);
   NodePtr sub_sub_graph_while_body_const_node = sub_sub_graph2->AddNode(sub_sub_graph_while_body_const_op_desc);*/
   OpDescPtr sub_sub_graph_while_body_data_op_desc = CreateOpDesc("body_data", DATA);
@@ -278,7 +278,7 @@ TEST_F(UtestGeHybrid, unfold_subgraphs_success) {
   ComputeGraphPtr sub_graph = std::make_shared<ComputeGraph>("sub_graph");
   OpDescPtr sub_graph_while_op_desc = CreateOpDesc("while", WHILE);
   NodePtr sub_graph_while_node = sub_graph->AddNode(sub_graph_while_op_desc);
-  sub_graph->SetGraphUnknownFlag(false);
+  sub_graph->SetGraphUnknownFlag(true);
   sub_graph_while_node->GetOpDesc()->AddSubgraphName("while_cond");
   sub_graph_while_node->GetOpDesc()->AddSubgraphName("while_body");
   sub_graph_while_node->GetOpDesc()->SetSubgraphInstanceName(0, "while_cond");
@@ -290,14 +290,16 @@ TEST_F(UtestGeHybrid, unfold_subgraphs_success) {
   partitioned_call_node->GetOpDesc()->AddSubgraphName("sub_graph");
   partitioned_call_node->GetOpDesc()->SetSubgraphInstanceName(0, "sub_graph");
 
-
   root_graph->AddSubGraph(sub_sub_graph1);
   root_graph->AddSubGraph(sub_sub_graph2);
   sub_sub_graph1->SetParentGraph(root_graph);
   sub_sub_graph2->SetParentGraph(root_graph);
+  sub_sub_graph1->SetParentNode(sub_graph_while_node);
+  sub_sub_graph2->SetParentNode(sub_graph_while_node);
 
   root_graph->AddSubGraph(sub_graph);
   sub_graph->SetParentNode(partitioned_call_node);
+  sub_graph->SetParentGraph(root_graph);
 
   GeRootModelPtr root_model = MakeShared<ge::GeRootModel>(root_graph);
   HybridModel hybrid_model(root_model);