diff --git a/tests/ut/ge/graph/build/graph_builder_unittest.cc b/tests/ut/ge/graph/build/graph_builder_unittest.cc
index f44c23a6..07615d3d 100644
--- a/tests/ut/ge/graph/build/graph_builder_unittest.cc
+++ b/tests/ut/ge/graph/build/graph_builder_unittest.cc
@@ -24,7 +24,7 @@
 #undef protected
 #undef private
 
-#include "graph/build/logical_stream_allocator.h"
+#include "graph/build/graph_builder.h"
 
 #include "common/types.h"
 #include "common/util.h"
@@ -101,8 +101,8 @@ TEST_F(UtestGraphBuilder, BuildForKnownShapeGraph_test) {
   ge::ComputeGraphPtr graph = make_shared<ge::ComputeGraph>("");
   MakeGraph(graph);
 
-  GraphBuilder builder;
-  GeModelPtr ge_model_ptr = nullptr;
+  ge::GraphBuilder builder;
+  ge::GeModelPtr ge_model_ptr = nullptr;
   builder.BuildForKnownShapeGraph(graph, ge_model_ptr, 0);
  }