diff --git a/tests/ut/ge/graph/load/kernel_ex_task_info_unittest.cc b/tests/ut/ge/graph/load/kernel_ex_task_info_unittest.cc
index b7693d76..68d029a8 100644
--- a/tests/ut/ge/graph/load/kernel_ex_task_info_unittest.cc
+++ b/tests/ut/ge/graph/load/kernel_ex_task_info_unittest.cc
@@ -140,7 +140,7 @@ TEST_F(UtestKernelExTaskInfo, kernel_ex_task_info_calculate_args) {
 TEST_F(UtestKernelExTaskInfo, kernel_ex_task_ext_info) {
   const string ext_info = {1, 1, 1, 1, 0, 0, 0, 0};
   const OpDescPtr op_desc = CreateOpDesc("FrameworkOp", "FrameworkOp");
-  AttrUtils::SetBool(op_desc, "_Allshape", true);
+  AttrUtils::SetBool(op_desc, "_AllShape", true);
 
   KernelExTaskInfo kernel_ex_task_info;
   EXPECT_EQ(kernel_ex_task_info.InitTaskExtInfo(ext_info, op_desc), SUCCESS);
diff --git a/tests/ut/ge/graph/load/kernel_task_info_unittest.cc b/tests/ut/ge/graph/load/kernel_task_info_unittest.cc
index a216160c..4fbfe61d 100644
--- a/tests/ut/ge/graph/load/kernel_task_info_unittest.cc
+++ b/tests/ut/ge/graph/load/kernel_task_info_unittest.cc
@@ -390,7 +390,7 @@ TEST_F(UtestKernelTaskInfo, init_kernel_taskInfo_with_aicpu_kernel_type_fail) {
   rtStreamCreate(&stream, 0);
   model.stream_list_ = { stream };
   model.op_list_[0] = CreateOpDesc("FrameworkOp", "FrameworkOp");
-  AttrUtils::SetBool(model.op_list_[0], "_Allshape", true);
+  AttrUtils::SetBool(model.op_list_[0], "_AllShape", true);
 
   domi::TaskDef task_def;
   KernelTaskInfo kernel_task_info;