diff --git a/tests/ut/ge/graph/build/graph_builder_unittest.cc b/tests/ut/ge/graph/build/graph_builder_unittest.cc
index 07615d3d..e4acb217 100644
--- a/tests/ut/ge/graph/build/graph_builder_unittest.cc
+++ b/tests/ut/ge/graph/build/graph_builder_unittest.cc
@@ -103,6 +103,6 @@ TEST_F(UtestGraphBuilder, BuildForKnownShapeGraph_test) {
 
   ge::GraphBuilder builder;
   ge::GeModelPtr ge_model_ptr = nullptr;
-  builder.BuildForKnownShapeGraph(graph, ge_model_ptr, 0);
+  builder.Build(graph, ge_model_ptr, 1);
  }