diff --git a/tests/ut/ge/graph/build/graph_mem_assigner_unittest.cc b/tests/ut/ge/graph/build/graph_mem_assigner_unittest.cc
index 983f1763..4bff9b38 100644
--- a/tests/ut/ge/graph/build/graph_mem_assigner_unittest.cc
+++ b/tests/ut/ge/graph/build/graph_mem_assigner_unittest.cc
@@ -30,7 +30,7 @@
 #define protected public
 #define private public
 #include "graph/build/memory/binary_block_mem_assigner.h"
-#include "graph/build/memory/graph_mem_assigner_unittest.h"
+#include "graph/build/memory/graph_mem_assigner.h"
 #include "graph/build/memory/hybrid_mem_assigner.h"
 #include "graph/build/memory/max_block_mem_assigner.h"
 #include "graph/manager/graph_var_manager.h"