diff --git a/tests/travis/run_test.sh b/tests/travis/run_test.sh
index 0e8b25e8a..ec41bc8a3 100755
--- a/tests/travis/run_test.sh
+++ b/tests/travis/run_test.sh
@@ -26,10 +26,14 @@ if [ ${TASK} == "python_test" ]; then
     echo "-------------------------------"
     source activate python3
     python --version
+    conda install numpy scipy pandas matplotlib nose scikit-learn
+    python -m pip install graphviz
     python -m nose tests/python || exit -1
     source activate python2
     echo "-------------------------------"
     python --version
+    conda install numpy scipy pandas matplotlib nose scikit-learn
+    python -m pip install graphviz
     python -m nose tests/python || exit -1
     exit 0
 fi
diff --git a/tests/travis/setup.sh b/tests/travis/setup.sh
index 03d71c90e..63f30207f 100755
--- a/tests/travis/setup.sh
+++ b/tests/travis/setup.sh
@@ -26,10 +26,4 @@ if [ ${TASK} == "python_test" ]; then
     conda info -a
     conda create -n python3 python=3.5
     conda create -n python2 python=2.7
-    source activate python3
-    conda install numpy scipy pandas matplotlib nose scikit-learn
-    python -m pip install graphviz
-    source activate python2
-    conda install numpy scipy pandas matplotlib nose scikit-learn
-    python -m pip install graphviz
 fi