diff --git a/python/orca/dev/example/run-example-tests.sh b/python/orca/dev/example/run-example-tests.sh
index fee45e09a2c..d2ebe126338 100644
--- a/python/orca/dev/example/run-example-tests.sh
+++ b/python/orca/dev/example/run-example-tests.sh
@@ -181,7 +181,7 @@ time5=$((now - start))
 echo "#6 start test for orca tf transfer_learning"
 #timer
 start=$(date "+%s")
-python ${BIGDL_ROOT}/python/orca/example/learn/tf/transfer_learning/transfer_learning.py --download_url $FTP_URI/analytics-zoo-data/cats_and_dogs_filtered.zip --epochs 1
+python ${BIGDL_ROOT}/python/orca/example/learn/tf/transfer_learning/transfer_learning.py --download_url $FTP_URI/analytics-zoo-data/cats_and_dogs_filtered.zip --epochs 1 --data_dir ./data	
 exit_status=$?
 if [ $exit_status -ne 0 ]; then
   echo "orca tf transfer_learning failed"