diff --git a/tests/ut/ge/hybrid/node_executor/aicpu/aicpu_node_executor_unittest.cc b/tests/ut/ge/hybrid/node_executor/aicpu/aicpu_node_executor_unittest.cc
index 153dda85..22f9b889 100644
--- a/tests/ut/ge/hybrid/node_executor/aicpu/aicpu_node_executor_unittest.cc
+++ b/tests/ut/ge/hybrid/node_executor/aicpu/aicpu_node_executor_unittest.cc
@@ -148,7 +148,7 @@ TEST_F(UtestAicpuNodeExecutor, aicpu_tf_node_task) {
 
   domi::TaskDef task_def2;
   task_def2.set_type(RT_MODEL_TASK_ALL_KERNEL);
-  domi::KernelExDef *kernel_def = task_def2.mutable_kernel();
+  domi::KernelDef *kernel_def = task_def2.mutable_kernel();
   kernel_def->set_args(reinterpret_cast<const char *>(&args), args.head.length);
   kernel_def->set_args_size(args.head.length);
   AicpuExtInfo aicpu_ext_info2;
@@ -169,7 +169,7 @@ TEST_F(UtestAicpuNodeExecutor, aicpu_tf_node_task) {
   //kernel_ex_def->set_allocated_kernel_ext_info(nullptr);
 
   free(ext_mem);
-
+  free(ext_mem2);
 }
 }  // namespace ge