diff --git a/python/paddle/fluid/contrib/slim/tests/test_post_training_quantization_mobilenetv1.py b/python/paddle/fluid/contrib/slim/tests/test_post_training_quantization_mobilenetv1.py
index 499367d339cab336d0e774cbfe30e75e1650997d..00a4e2c2aa49e03f7dfb41bca640624d6862f84c 100644
--- a/python/paddle/fluid/contrib/slim/tests/test_post_training_quantization_mobilenetv1.py
+++ b/python/paddle/fluid/contrib/slim/tests/test_post_training_quantization_mobilenetv1.py
@@ -242,7 +242,6 @@ class TestPostTrainingQuantization(unittest.TestCase):
                                  model_path,
                                  quantizable_op_type,
                                  batch_size,
-                                 batch_nums,
                                  algo="KL",
                                  round_type="round",
                                  is_full_quantize=False,
@@ -268,7 +267,6 @@ class TestPostTrainingQuantization(unittest.TestCase):
                                        batch_size=batch_size,
                                        batch_nums=batch_nums,
                                        algo=algo,
-                                       batch_nums=batch_nums,
                                        quantizable_op_type=quantizable_op_type,
                                        round_type=round_type,
                                        is_full_quantize=is_full_quantize,