diff --git a/python/paddle/fluid/tests/unittests/ps/CMakeLists.txt b/python/paddle/fluid/tests/unittests/ps/CMakeLists.txt
index fb868e82ce7cfbd6c50de4be7cf54f1d80d3117e..24960e14a014a3618f90574a60355077ee974564 100755
--- a/python/paddle/fluid/tests/unittests/ps/CMakeLists.txt
+++ b/python/paddle/fluid/tests/unittests/ps/CMakeLists.txt
@@ -4,9 +4,7 @@ file(
   "test_*.py")
 string(REPLACE ".py" "" TEST_OPS "${TEST_OPS}")
 
-if(NOT WITH_HETERPS)
-  list(REMOVE_ITEM TEST_OPS "test_gpubox_ps")
-endif()
+list(REMOVE_ITEM TEST_OPS "test_gpubox_ps")
 
 foreach(TEST_OP ${TEST_OPS})
   py_test_modules(${TEST_OP} MODULES ${TEST_OP})