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 61427594..402170c3 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
@@ -201,6 +201,7 @@ TEST_F(UtestAicpuNodeExecutor, aicpu_memcopy_task) {
   ASSERT_EQ(aicpu_node_task.SetMemCopyTask(task_def), SUCCESS);
   node_item->num_outputs = 1;
   AicpuNodeTask aicpu_node_task2(node_item, task_def);
+  ASSERT_EQ(aicpu_node_task2.SetMemCopyTask(task_def), FAILED);
   kernel_def->set_args_size(0);
   ASSERT_EQ(aicpu_node_task2.SetMemCopyTask(task_def), FAILED);
   char* args2 = "123";
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 a6466972..666c31c8 100644
--- a/tests/ut/ge/single_op/single_op_model_unittest.cc
+++ b/tests/ut/ge/single_op/single_op_model_unittest.cc
@@ -424,6 +424,8 @@ TEST_F(UtestSingleOpModel, build_memcpoy_task) {
   args.head.ioAddrNum = 6;
   domi::KernelDef kernel_def;
   kernel_def.set_args(reinterpret_cast<const char *>(&args), args.head.length);
+  kernel_def.set_args_size(args.head.length);
+  ASSERT_EQ(aicpu_task.SetMemCopyTask(kernel_def), FAILED);
   kernel_def.set_args_size(0);
   AiCpuCCTask aicpu_task;
   ASSERT_EQ(aicpu_task.SetMemCopyTask(kernel_def), FAILED);