diff --git a/scripts/build_wheel.sh b/scripts/build_wheel.sh
index 6fa9c0b20b7bf622543a8abc42a76bdb722a20e8..c3445cd750647629c36131fb4b95981713ba02d8 100644
--- a/scripts/build_wheel.sh
+++ b/scripts/build_wheel.sh
@@ -91,9 +91,8 @@ function unittest() {
   if [ $? != 0  ]; then
     exit 1
   fi
-  find "../ppdet" -name 'tests' -type d -print0 | \
-      xargs -0 -I{} -n1 bash -c \
-      'python -m unittest discover -v -s {}'
+  find "../ppdet" -wholename '*tests/test_*' -type f -print0 | \
+      xargs -0 -I{} -n1 -t bash -c  'python -u -s {}'
 
   # clean TEST_DIR
   cd ..