diff --git a/tests/ut/ge/hybrid/ge_hybrid_unittest.cc b/tests/ut/ge/hybrid/ge_hybrid_unittest.cc
index 1b2223b3..eac2da33 100644
--- a/tests/ut/ge/hybrid/ge_hybrid_unittest.cc
+++ b/tests/ut/ge/hybrid/ge_hybrid_unittest.cc
@@ -260,13 +260,10 @@ TEST_F(UtestGeHybrid, hybrid_model_executor) {
 TEST_F(UtestGeHybrid, unfold_subgraphs_success) {
   ComputeGraphPtr merged_graph = nullptr;
 
-  ComputeGraphPtr root_graph = std::make_shared<ComputeGraph>("root_graph");
-  auto partitioned_call_op_desc = MakeShared<OpDesc>("partitioned_call", PARTITIONEDCALL);
-  auto partitioned_call_node = root_graph->AddNode(partitioned_call_op_desc);
-
   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(True);
 
   ComputeGraphPtr sub_sub_graph1 = std::make_shared<ComputeGraph>("while_cond");
   OpDescPtr sub_sub_graph_while_cond_data_op_desc = CreateOpDesc("cond_data", DATA);
@@ -282,6 +279,11 @@ TEST_F(UtestGeHybrid, unfold_subgraphs_success) {
   sub_sub_graph_while_body_add_node->AddLinkFrom(sub_sub_graph_while_body_data_node);
   sub_sub_graph_while_body_add_node->AddLinkFrom(sub_sub_graph_while_body_const_node);
 
+  ComputeGraphPtr root_graph = std::make_shared<ComputeGraph>("root_graph");
+  auto partitioned_call_op_desc = MakeShared<OpDesc>("partitioned_call", PARTITIONEDCALL);
+  auto partitioned_call_node = root_graph->AddNode(partitioned_call_op_desc);
+  partitioned_call_node->GetOpDesc()->AddSubgraphName("sub_graph");
+
   sub_sub_graph1->AddSubGraph(sub_sub_graph1);
   sub_sub_graph2->AddSubGraph(sub_sub_graph2);