diff --git a/python/paddle/fluid/tests/unittests/test_dist_transpiler.py b/python/paddle/fluid/tests/unittests/test_dist_transpiler.py
index 73795a21549e5d0461553d2df3a715881142c163..0555db4cba4f598f25a226439a5ec8edf6e5c928 100644
--- a/python/paddle/fluid/tests/unittests/test_dist_transpiler.py
+++ b/python/paddle/fluid/tests/unittests/test_dist_transpiler.py
@@ -871,8 +871,8 @@ class TestRemoteNce(TestDistLookupTableBase):
     def transpiler_test_impl(self):
         trainer, _ = self.get_trainer()
 
-        out_vars = ["nce_w.block0", "nce_w.block1"]
-        in_vars = ["nce_b.block0", "nce_b.block1"]
+        out_vars = ["nce_w"]
+        in_vars = ["nce_b"]
 
         recv_var_names = []