diff --git a/python/paddle/fluid/tests/unittests/test_nce_remote_table_op.py b/python/paddle/fluid/tests/unittests/test_nce_remote_table_op.py
index e87545cb9c67e041c0d7225e1ddc1b6aca257be4..5e440bf35d2a46a63ca40e69e7dd99df1a1fbe4c 100644
--- a/python/paddle/fluid/tests/unittests/test_nce_remote_table_op.py
+++ b/python/paddle/fluid/tests/unittests/test_nce_remote_table_op.py
@@ -141,6 +141,7 @@ class TestListenAndServOp(unittest.TestCase):
                     SampleLabels='SampleLabels',
                     num_total_classes=5,
                     num_neg_samples=2,
+                    custom_neg_classes=list(range(2)),
                     sampler=0,
                     seed=1,
                     is_sparse=True,