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 b225949b..1b15914b 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
@@ -151,18 +151,16 @@ TEST_F(UtestAicpuNodeExecutor, aicpu_tf_node_task) {
   task_def2.mutable_kernel()->set_args(reinterpret_cast<const char *>(&args), args.head.length);
   task_def2.mutable_kernel()->set_args_size(args.head.length);
 
-  hybrid_model.task_defs_[node] = std::vector<domi::TaskDef>({task_def2});
+  hybrid_model.task_defs_[node] = std::vector<domi::TaskDef>({task_def2, task_def2});
 
   AicpuNodeTask aicpu_node_task(node_item, task_def);
-  ASSERT_EQ(aicpu_node_task.Init(hybrid_model), FAILED);
+  ASSERT_EQ(aicpu_node_task.Init(hybrid_model), SUCCESS);
   ASSERT_EQ(aicpu_node_task.LaunchTask(*node_state->GetTaskContext()), SUCCESS);
 
-
   //kernel_ex_def->set_allocated_kernel_ext_info(nullptr);
 
   free(ext_mem);
 
 }
-
 }  // namespace ge