diff --git a/python/paddle/fluid/tests/unittests/ps/test_gpubox_ps.py b/python/paddle/fluid/tests/unittests/ps/test_gpubox_ps.py
index 7c7fedd812c6d12c75d23fab22fbbc5763574ef8..4562561fce42afd9c6081a086abc96920d02df36 100755
--- a/python/paddle/fluid/tests/unittests/ps/test_gpubox_ps.py
+++ b/python/paddle/fluid/tests/unittests/ps/test_gpubox_ps.py
@@ -27,8 +27,9 @@ class GpuBoxTest(unittest.TestCase):
         exitcode = os.system('bash gpubox_run.sh')
         if os.path.exists('./train_data'):
             os.system('rm -rf train_data')
-        if os.path.exists('./log'):
-            os.system('rm -rf log')
+        if exitcode:
+            os.system('cat ./log/worker.0.log')
+
         assert exitcode == 0
 
 
diff --git a/python/paddle/incubate/fleet/parameter_server/distribute_transpiler/__init__.py b/python/paddle/incubate/fleet/parameter_server/distribute_transpiler/__init__.py
index 99e5dd2219e37d1da9a141a41991c28a0c3a0332..fa1da4bcf0b68b3915f33c156d35de02828feeb2 100644
--- a/python/paddle/incubate/fleet/parameter_server/distribute_transpiler/__init__.py
+++ b/python/paddle/incubate/fleet/parameter_server/distribute_transpiler/__init__.py
@@ -57,8 +57,9 @@ from paddle.incubate.fleet.parameter_server.distribute_transpiler.distributed_st
     StrategyFactory,
 )
 
-from paddle.fluid.transpiler.details.checkport import wait_server_ready
-
+from paddle.distributed.fleet.base.private_helper_function import (
+    wait_server_ready,
+)
 from paddle.fluid.incubate.fleet.parameter_server.mode import PSMode
 from paddle.fluid.incubate.fleet.base.fleet_base import DistributedOptimizer