diff --git a/ge/single_op/single_op_model.cc b/ge/single_op/single_op_model.cc
index a0b0d608..2439a82b 100755
--- a/ge/single_op/single_op_model.cc
+++ b/ge/single_op/single_op_model.cc
@@ -279,7 +279,7 @@ Status SingleOpModel::BuildTaskList(SingleOp &single_op) {
       if (ret != SUCCESS) {
         return ret;
       }
-      string op_name = op_list_[context.op_index()]->GetName();
+      string op_name = op_list_[task_def.kernel_ex().op_index()]->GetName();
       single_op.tasks_.emplace_back(aicpu_task);
       single_op.op_name_.emplace_back(op_name);
     } else {