diff --git a/paddle/scripts/paddle_build.sh b/paddle/scripts/paddle_build.sh
index a55a9e89f7613faa64999529bdb09915655cd6ad..02bf8533d86dd138d9450fb1a3cedb70cce8a964 100755
--- a/paddle/scripts/paddle_build.sh
+++ b/paddle/scripts/paddle_build.sh
@@ -315,7 +315,8 @@ function run_test() {
 EOF
         echo $http_proxy
         echo $https_proxy
-        ctest -V
+        ctest -R test_parallel_executor_fetch_feed -V
+        ctest -R test_dist_se_resnext -V
         # make install should also be test when unittest
         make install -j `nproc`
         pip install /usr/local/opt/paddle/share/wheels/*.whl
diff --git a/python/paddle/dataset/flowers.py b/python/paddle/dataset/flowers.py
index 8c9c721b33c7e4de4781ac76ceda3fd3a141d3d6..c4a3eb55dd16cdf7962a0718f51df3fce9a1aa65 100644
--- a/python/paddle/dataset/flowers.py
+++ b/python/paddle/dataset/flowers.py
@@ -41,6 +41,7 @@ from paddle.reader import *
 import os
 import numpy as np
 from multiprocessing import cpu_count
+import six
 from six.moves import cPickle as pickle
 from six.moves import zip
 __all__ = ['train', 'test', 'valid']
diff --git a/python/paddle/fluid/tests/unittests/CMakeLists.txt b/python/paddle/fluid/tests/unittests/CMakeLists.txt
index 0c9bbb766f0cdb1eacf80aaa263ea0b6c4dcc6a3..228a5ab917b30f62d15a5a27b3502a6b4437157e 100644
--- a/python/paddle/fluid/tests/unittests/CMakeLists.txt
+++ b/python/paddle/fluid/tests/unittests/CMakeLists.txt
@@ -64,7 +64,7 @@ if(WITH_DISTRIBUTE)
 endif()
 py_test_modules(test_parallel_executor_crf MODULES test_parallel_executor_crf SERIAL)
 py_test_modules(test_parallel_executor_fetch_feed MODULES test_parallel_executor_fetch_feed SERIAL)
-set_tests_properties(test_parallel_executor_fetch_feed PROPERTIES TIMEOUT 600)
+set_tests_properties(test_parallel_executor_fetch_feed PROPERTIES TIMEOUT 300)
 py_test_modules(test_dist_transformer MODULES test_dist_transformer SERIAL)
 py_test_modules(test_dist_se_resnext MODULES test_dist_se_resnext SERIAL)
 py_test_modules(test_parallel_executor_transformer MODULES test_parallel_executor_transformer SERIAL)