diff --git a/tests/ut/ge/graph/build/graph_builder_unittest.cc b/tests/ut/ge/graph/build/graph_builder_unittest.cc
index e4acb217..02726af1 100644
--- a/tests/ut/ge/graph/build/graph_builder_unittest.cc
+++ b/tests/ut/ge/graph/build/graph_builder_unittest.cc
@@ -102,7 +102,7 @@ TEST_F(UtestGraphBuilder, BuildForKnownShapeGraph_test) {
   MakeGraph(graph);
 
   ge::GraphBuilder builder;
-  ge::GeModelPtr ge_model_ptr = nullptr;
-  builder.Build(graph, ge_model_ptr, 1);
+  ge::GeRootModelPtr ge_root_model = nullptr;
+  builder.Build(graph, ge_root_model, 1);
  }