diff --git a/.travis.yml b/.travis.yml
index 61d31132b73be28fac43f8b5b5248b7390caca82..27a4be38f5456ce065bafe2109e6ab5c199c493a 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -25,7 +25,9 @@ addons:
       - git
       - build-essential
       - libatlas-base-dev
+      - python
       - python-pip
+      - python2.7-dev
       - curl
       - graphviz
       - clang-format-3.8
diff --git a/paddle/api/CMakeLists.txt b/paddle/api/CMakeLists.txt
index dd617e32681e77e15296b476a781fc9e28d56b4f..3ac50e34bb434b14d346f1c4707084f93461284d 100644
--- a/paddle/api/CMakeLists.txt
+++ b/paddle/api/CMakeLists.txt
@@ -94,17 +94,17 @@ add_dependencies(python_api_wheel python_swig_sources
   paddle_cuda)
 
 if(WITH_TESTING)
-  SET(PIP_SOURCES_DIR ${PYTHON_SOURCES_DIR}/pip)
-  ExternalProject_Add(pip
-      ${EXTERNAL_PROJECT_LOG_ARGS}
-      GIT_REPOSITORY      https://github.com/pypa/pip.git
-      GIT_TAG             9.0.1
-      PREFIX              ${PIP_SOURCES_DIR}
-      CONFIGURE_COMMAND   ""
-      BUILD_COMMAND       ""
-      INSTALL_COMMAND     env ${py_env} ${PYTHON_EXECUTABLE} setup.py install
-      BUILD_IN_SOURCE     1
-      DEPENDS python setuptools python_api_wheel
-  )
+    SET(PIP_SOURCES_DIR ${PYTHON_SOURCES_DIR}/pip)
+    ExternalProject_Add(pip
+        ${EXTERNAL_PROJECT_LOG_ARGS}
+        GIT_REPOSITORY      https://github.com/pypa/pip.git
+        GIT_TAG             9.0.1
+        PREFIX              ${PIP_SOURCES_DIR}
+        CONFIGURE_COMMAND   ""
+        BUILD_COMMAND       ""
+        INSTALL_COMMAND     env ${py_env} ${PYTHON_EXECUTABLE} setup.py install
+        BUILD_IN_SOURCE     1
+        DEPENDS python setuptools python_api_wheel
+    )
     add_subdirectory(test)
 endif()
diff --git a/paddle/api/test/CMakeLists.txt b/paddle/api/test/CMakeLists.txt
index 985df6f56ea528564f20dee6554593b341af5331..a2fa623c80087d42e6a2a5c05f62eba4997f8ec4 100644
--- a/paddle/api/test/CMakeLists.txt
+++ b/paddle/api/test/CMakeLists.txt
@@ -1,2 +1,2 @@
 add_test(NAME test_swig_api
-    COMMAND bash ${PROJ_ROOT}/paddle/api/test/run_tests.sh ${PYTHON_EXECUTABLE} ${PYTHON_INSTALL_DIR}/bin/pip)
+    COMMAND bash ${PROJ_ROOT}/paddle/api/test/run_tests.sh ${PYTHON_EXECUTABLE})
diff --git a/paddle/api/test/run_tests.sh b/paddle/api/test/run_tests.sh
index f00ec2c967d8c45984ee9cbdebdc3ee2a314d267..bcf06afa86aaa1a3151aeb966b54f69657c541e3 100755
--- a/paddle/api/test/run_tests.sh
+++ b/paddle/api/test/run_tests.sh
@@ -20,10 +20,6 @@ popd > /dev/null
 
 cd $SCRIPTPATH
 
-# rm -rf .test_env
-# virtualenv .test_env
-# source .test_env/bin/activate
-
 $1 -m pip install ../../dist/*.whl
 
 test_list="testArguments.py testGradientMachine.py testMatrix.py  testVector.py testTrain.py testTrainer.py"