diff --git a/tests/ut/common/graph/testcase/ge_graph/ge_tensor_unittest.cc b/tests/ut/common/graph/testcase/ge_graph/ge_tensor_unittest.cc
index 6d34ab59..5c75bd01 100644
--- a/tests/ut/common/graph/testcase/ge_graph/ge_tensor_unittest.cc
+++ b/tests/ut/common/graph/testcase/ge_graph/ge_tensor_unittest.cc
@@ -230,7 +230,7 @@ TEST_F(UtestGeTensor, test_tensor_invalid_null) {
   GeTensor tensor(msg_owner, nullptr);
   EXPECT_EQ(tensor.GetData().size(), 0);
   EXPECT_EQ(tensor.MutableData().size(), 0);
-  EXPECT_EQ(tensor.SetData(Buffer(100)), ge::GRAPH_PARAM_INVALID);
+  EXPECT_EQ(tensor.SetData(Buffer(100)), GRAPH_SUCCESS);
 
   TensorUtils::SetWeightSize(tensor.MutableTensorDesc(), 100);
   EXPECT_EQ(TensorUtils::GetWeightSize(tensor), 0);