diff --git a/tests/ut/ge/generator/ge_generator_unittest.cc b/tests/ut/ge/generator/ge_generator_unittest.cc
index 5cf5038b..066577df 100644
--- a/tests/ut/ge/generator/ge_generator_unittest.cc
+++ b/tests/ut/ge/generator/ge_generator_unittest.cc
@@ -76,7 +76,7 @@ TEST_F(UtestGeGenerator, test_build_single_op_online) {
 TEST_F(UtestGeGenerator, test_set_model_name) {
   GeGenerator generator;
   generator.Initialize({});
-  GeRootModelPtr ge_root_model = make_shared<GeRootModelPtr>(GeRootModel());
+  GeRootModelPtr ge_root_model = make_shared<GeRootModel>(GeRootModel());
   ComputeGraphPtr graph = make_shared<ComputeGraph>(ComputeGraph("graph"));
   (void)AttrUtils::SetBool(graph, "_dynamic_shape_partitioned", true);
   ge_root_model->root_graph_ = std::move(graph);