diff --git a/cmake/generic.cmake b/cmake/generic.cmake
index 957c20bcf603f2f264b4658f63ac0eec438f12b1..9f907a9dc296900ce1c0fab7fa87b0f558f5d758 100644
--- a/cmake/generic.cmake
+++ b/cmake/generic.cmake
@@ -411,7 +411,7 @@ function(py_test TARGET_NAME)
     set(multiValueArgs SRCS DEPS)
     cmake_parse_arguments(py_test "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})  
     add_test(NAME ${TARGET_NAME}
-             COMMAND env PYTHONPATH=${PADDLE_PYTHON_PACKAGE_DIR}
+             COMMAND env PYTHONPATH=${PADDLE_PYTHON_LIB_DIR}
              python2 ${py_test_SRCS}
              WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR})
   endif()
diff --git a/python/CMakeLists.txt b/python/CMakeLists.txt
index b5030da8e75eb94e857ae4effc6adb6d19dc0e93..fc8c6f6a42c530a28d444e639d777d0e86f7573a 100644
--- a/python/CMakeLists.txt
+++ b/python/CMakeLists.txt
@@ -1,6 +1,8 @@
 set(OUTPUT_DIR
     "${CMAKE_CURRENT_BINARY_DIR}/build")
 
+set(PADDLE_PYTHON_LIB_DIR "${OUTPUT_DIR}/lib")
+
 file(GLOB TRAINER_PY_FILES . ./paddle/trainer/*.py)
 file(GLOB HELPERS_PY_FILES . ./paddle/trainer_config_helpers/*.py)
 file(GLOB UTILS_PY_FILES . ./paddle/utils/*.py)