diff --git a/ge/single_op/task/op_task.h b/ge/single_op/task/op_task.h
index f93e031a..97d8a342 100644
--- a/ge/single_op/task/op_task.h
+++ b/ge/single_op/task/op_task.h
@@ -124,7 +124,7 @@ class TbeOpTask : public OpTask {
   std::string original_kernel_key_;
   std::string node_info_;
   std::vector<size_t> arg_index_; // data index in args
-  size_t input_num_; // Include const input
+  size_t input_num_; // include const input
   size_t output_num_;
 };
 
diff --git a/tests/ut/ge/single_op/single_op_task_unittest.cc b/tests/ut/ge/single_op/single_op_task_unittest.cc
index 020efc23..b0c98205 100644
--- a/tests/ut/ge/single_op/single_op_task_unittest.cc
+++ b/tests/ut/ge/single_op/single_op_task_unittest.cc
@@ -134,6 +134,8 @@ TEST_F(UtestSingleOpTask, test_update_ioaddr) {
   task.arg_size_ = sizeof(void *) * 4;
   task.args_.reset(new (std::nothrow) uint8_t[task.arg_size_]);
   task.arg_index_ = {0};
+  task.input_num_ = 2;
+  task.output_num_ = 1;
 
   vector<void *> args;
   vector<DataBuffer> inputs;