diff --git a/tests/ut/ge/graph/load/kernel_task_info_unittest.cc b/tests/ut/ge/graph/load/kernel_task_info_unittest.cc
index 8a255f16..2cfb2a76 100644
--- a/tests/ut/ge/graph/load/kernel_task_info_unittest.cc
+++ b/tests/ut/ge/graph/load/kernel_task_info_unittest.cc
@@ -720,6 +720,7 @@ TEST_F(UtestKernelTaskInfo, success_kernel_taskInfo_init_set_context) {
   context->set_args_count(1);
   context->set_args_offset("args111111", 10);
 
+  kernel_task_info.op_desc_ = CreateOpDesc("FrameworkOp", "FrameworkOp");
   EXPECT_EQ(kernel_task_info.SetContext(*kernel_def), SUCCESS);
 
   EXPECT_EQ(kernel_task_info.Release(), SUCCESS);
@@ -741,6 +742,7 @@ TEST_F(UtestKernelTaskInfo, kernel_taskInfo_init_set_context_failed1) {
   context->set_is_flowtable(true);
   context->set_args_count(0);
 
+  kernel_task_info.op_desc_ = CreateOpDesc("FrameworkOp", "FrameworkOp");
   EXPECT_EQ(kernel_task_info.SetContext(*kernel_def), INTERNAL_ERROR);
 
   kernel_def->clear_context();
@@ -760,6 +762,8 @@ TEST_F(UtestKernelTaskInfo, kernel_taskInfo_init_set_context_failed2) {
   context->set_args_count(5);
   context->set_args_offset("\0\0");  // args_offset = 0
 
+  kernel_task_info.op_desc_ = CreateOpDesc("FrameworkOp", "FrameworkOp");
+
   EXPECT_EQ(kernel_task_info.SetContext(*kernel_def), PARAM_INVALID);
 
   kernel_def->clear_context();