diff --git a/ppdet/modeling/tests/test_mstest.py b/ppdet/modeling/tests/test_mstest.py
index 57d1d169fecbf235227aa388ae0d2e3d92330d0c..a5b75110afd326f6bff10acdc85eb4b0461ba910 100644
--- a/ppdet/modeling/tests/test_mstest.py
+++ b/ppdet/modeling/tests/test_mstest.py
@@ -21,6 +21,7 @@ import unittest
 from ppdet.core.workspace import load_config
 from ppdet.engine import Trainer
 
+
 class TestMultiScaleInference(unittest.TestCase):
     def setUp(self):
         self.set_config()
@@ -48,12 +49,13 @@ class TestMultiScaleInference(unittest.TestCase):
         tests_img_root = os.path.join(os.path.dirname(__file__), 'imgs')
 
         # input images to predict
-        imgs = ['coco2017_val2017_000000000139.jpg', 'coco2017_val2017_000000000724.jpg']
+        imgs = [
+            'coco2017_val2017_000000000139.jpg',
+            'coco2017_val2017_000000000724.jpg'
+        ]
         imgs = [os.path.join(tests_img_root, img) for img in imgs]
-        trainer.predict(imgs,
-                        draw_threshold=0.5,
-                        output_dir='output',
-                        save_txt=True)
+        trainer.predict(
+            imgs, draw_threshold=0.5, output_dir='output', save_results=False)
 
 
 if __name__ == '__main__':
diff --git a/scripts/build_wheel.sh b/scripts/build_wheel.sh
index c3445cd750647629c36131fb4b95981713ba02d8..8b6ea06836b30328e2b54e916c4f6ba70834264c 100644
--- a/scripts/build_wheel.sh
+++ b/scripts/build_wheel.sh
@@ -26,6 +26,7 @@ EGG_DIR="paddledet.egg-info"
 
 CFG_DIR="configs"
 TEST_DIR=".tests"
+DATA_DIR="dataset"
 
 # command line log config
 RED='\033[0;31m'
@@ -86,6 +87,7 @@ function unittest() {
   #       make sure installed paddledet is used
   mkdir $TEST_DIR
   cp -r $CFG_DIR $TEST_DIR
+  cp -r $DATA_DIR $TEST_DIR
   cd $TEST_DIR
 
   if [ $? != 0  ]; then