diff --git a/tests/ut/ge/single_op/single_op_model_unittest.cc b/tests/ut/ge/single_op/single_op_model_unittest.cc
index 77f3e3f1..a6466972 100644
--- a/tests/ut/ge/single_op/single_op_model_unittest.cc
+++ b/tests/ut/ge/single_op/single_op_model_unittest.cc
@@ -422,15 +422,13 @@ TEST_F(UtestSingleOpModel, build_memcpoy_task) {
   AicpuTaskStruct args;
   args.head.length = sizeof(args);
   args.head.ioAddrNum = 6;
-  domi::TaskDef *task_def = model_task_def->add_task();
-  task_def->set_type(RT_MODEL_TASK_KERNEL);
-  domi::KernelDef *kernel_def = task_def->mutable_kernel();
-  kernel_def->set_args(reinterpret_cast<const char *>(&args), args.head.length);
-  kernel_def->set_args_size(0);
+  domi::KernelDef kernel_def;
+  kernel_def.set_args(reinterpret_cast<const char *>(&args), args.head.length);
+  kernel_def.set_args_size(0);
   AiCpuCCTask aicpu_task;
   ASSERT_EQ(aicpu_task.SetMemCopyTask(kernel_def), FAILED);
   char* args2 = "123";
-  kernel_def->set_args(reinterpret_cast<const char *>(&args2), 3);
-  kernel_def->set_args_size(3);
+  kernel_def.set_args(reinterpret_cast<const char *>(&args2), 3);
+  kernel_def.set_args_size(3);
   ASSERT_EQ(aicpu_task.SetMemCopyTask(kernel_def), FAILED);
 }