diff --git a/.azure-pipelines/scripts/ut/compare_coverage.sh b/.azure-pipelines/scripts/ut/compare_coverage.sh
index 9ddb0e932b0..006c44c0894 100644
--- a/.azure-pipelines/scripts/ut/compare_coverage.sh
+++ b/.azure-pipelines/scripts/ut/compare_coverage.sh
@@ -229,7 +229,7 @@ function main {
             done
         fi
     done
-    # generage table end
+    # generate table end
     echo """</table></div></body></html>""" >> ${output_file}
 
 }
diff --git a/.azure-pipelines/scripts/ut/env_setup.sh b/.azure-pipelines/scripts/ut/env_setup.sh
index f6394c3f198..07fa00a8d35 100644
--- a/.azure-pipelines/scripts/ut/env_setup.sh
+++ b/.azure-pipelines/scripts/ut/env_setup.sh
@@ -4,8 +4,9 @@ set -x
 echo "copy pre-train model..."
 mkdir -p /tmp/.neural_compressor/inc_ut || true
 cp -r /tf_dataset/ut-localfile/resnet_v2 /tmp/.neural_compressor/inc_ut || true
-mkdir -p /root/.keras/datasets || true
-cp -r /tf_dataset/ut-localfile/cifar-10-batches-py /root/.keras/datasets || true
+mkdir -p ~/.keras/datasets || true
+cp -r /tf_dataset/ut-localfile/cifar-10-batches-py* ~/.keras/datasets || true
+ll ~/.keras/datasets
 
 echo "install dependencies..."
 echo "tensorflow version is $tensorflow_version"