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 adf11e815faa97bb547b9b1d50f839baa56e9595..12dc47785d262b7685ff14300e41e15e275eddb2 100644
--- a/python/paddle/fluid/tests/unittests/test_cross_entropy_loss.py
+++ b/python/paddle/fluid/tests/unittests/test_cross_entropy_loss.py
@@ -13,6 +13,7 @@
 # limitations under the License.
 
 from __future__ import print_function
+from ast import Index
 
 import paddle
 import paddle.fluid as fluid
@@ -1477,7 +1478,7 @@ class TestCrossEntropyFAPIError(unittest.TestCase):
                     weight=weight_data,
                     ignore_index=-100)
 
-            self.assertRaises(ValueError, test_LabelValue_ExceedMax)
+            self.assertRaises(IndexError, test_LabelValue_ExceedMax)
 
             def test_LabelValue_ExceedMin():
                 input_data = paddle.rand(shape=[20, 100])
@@ -1491,7 +1492,7 @@ class TestCrossEntropyFAPIError(unittest.TestCase):
                     weight=weight_data,
                     ignore_index=-100)
 
-            self.assertRaises(ValueError, test_LabelValue_ExceedMin)
+            self.assertRaises(IndexError, test_LabelValue_ExceedMin)
 
             def static_test_WeightLength_NotEqual():
                 input_np = np.random.random([2, 4]).astype('float32')