diff --git a/python/paddle/fluid/tests/unittests/CMakeLists.txt b/python/paddle/fluid/tests/unittests/CMakeLists.txt
index c45964715956d2cf1c35455f8eec3e71a9f3d982..4ac3fff5255c501300b1307af266c5a1b334f1de 100644
--- a/python/paddle/fluid/tests/unittests/CMakeLists.txt
+++ b/python/paddle/fluid/tests/unittests/CMakeLists.txt
@@ -302,7 +302,7 @@ if(WITH_DISTRIBUTE)
             # NOTE. test_launch only work in gpu collective mode
             bash_test_modules(test_launch MODULES test_launch.sh  ENVS PADDLE_BINARY_DIR=${PADDLE_BINARY_DIR})
         endif()
-        bash_test_modules(test_launch_ps MODULES test_launch_ps.sh)
+        bash_test_modules(test_launch_ps MODULES test_launch_ps.sh ENVS PADDLE_BINARY_DIR=${PADDLE_BINARY_DIR})
 
         set(dist_ut_port 1000)
         foreach(TEST_OP ${DIST_TEST_OPS})
diff --git a/python/paddle/fluid/tests/unittests/test_launch_ps.sh b/python/paddle/fluid/tests/unittests/test_launch_ps.sh
index 0bd722af03aae743fc960adda1e4f71bfb0a11d9..78452b5fe37ff686d58ce5f529e7db087b882ad6 100644
--- a/python/paddle/fluid/tests/unittests/test_launch_ps.sh
+++ b/python/paddle/fluid/tests/unittests/test_launch_ps.sh
@@ -1,7 +1,8 @@
 #!/bin/bash
 set -e
 # use default values
-python -m paddle.distributed.launch_ps fleet_ps_training.py 2> ut.elog
+launch_py=${PADDLE_BINARY_DIR}/python/paddle/distributed/launch_ps.py
+python ${launch_py} fleet_ps_training.py 2> ut.elog
 
 if grep -q "server are killed" ut.elog; then
     echo "succeed"