diff --git a/.azure-pipelines/template/ut-template.yml b/.azure-pipelines/template/ut-template.yml
index 4efec29b..f31eecbe 100644
--- a/.azure-pipelines/template/ut-template.yml
+++ b/.azure-pipelines/template/ut-template.yml
@@ -43,6 +43,7 @@ steps:
   - ${{ if eq(parameters.imageSource, 'pull') }}:
     - script: |
         docker exec ${{ parameters.utContainerName }} bash -c "cd /auto-round \
+          && python setup.py bdist_wheel \
           && pip install $(ls dist/*.whl | head -n 1)[hpu] \
           && pip list"
       displayName: "HPU Env Setup"