diff --git a/python/paddle/fluid/tests/unittests/test_cross_entropy_loss.py b/python/paddle/fluid/tests/unittests/test_cross_entropy_loss.py
index f99c3b2129b1a9a68fb65d03d6de40736a96d3e8..e9116ae1b44346cc3ea13daa58513cf24a14120a 100644
--- a/python/paddle/fluid/tests/unittests/test_cross_entropy_loss.py
+++ b/python/paddle/fluid/tests/unittests/test_cross_entropy_loss.py
@@ -1373,8 +1373,9 @@ class TestCrossEntropyFAPIError(unittest.TestCase):
                 label_data = paddle.randint(
                     0, 100, shape=[20, 1], dtype="int64")
                 label_data[0] = 255
+                weight_data = paddle.rand([100])
                 paddle.nn.functional.cross_entropy(
-                    input=input_data, label=label_data)
+                    input=input_data, label=label_data, weight=weight_data)
 
             self.assertRaises(ValueError, test_LabelValue)
 
@@ -1383,8 +1384,9 @@ class TestCrossEntropyFAPIError(unittest.TestCase):
                 label_data = paddle.randint(
                     0, 100, shape=[20, 1], dtype="int64")
                 label_data[0] = -1
+                weight_data = paddle.rand([100])
                 paddle.nn.functional.cross_entropy(
-                    input=input_data, label=label_data)
+                    input=input_data, label=label_data, weight=weight_data)
 
             self.assertRaises(ValueError, test_LabelValueNeg)