diff --git a/.azure-pipelines/template/ut-template.yml b/.azure-pipelines/template/ut-template.yml
index 76c11142..a1b84a7e 100644
--- a/.azure-pipelines/template/ut-template.yml
+++ b/.azure-pipelines/template/ut-template.yml
@@ -43,7 +43,8 @@ steps:
   - ${{ if eq(parameters.imageSource, 'pull') }}:
     - script: |
         docker exec ${{ parameters.utContainerName }} bash -c "cd /auto-round \
-          && pip install .[hpu] \
+          && python setup.py bdist_wheel lib \
+          && pip install dist/*.whl \
           && pip list"
       displayName: "HPU Env Setup"
 
diff --git a/MANIFEST.in b/MANIFEST.in
index 906cf0d2..471010b7 100644
--- a/MANIFEST.in
+++ b/MANIFEST.in
@@ -1,4 +1,4 @@
 include requirements.txt
 include requirements-cpu.txt
 include requirements-hpu.txt
-include requirements-gpu.txt
+include requirements-lib.txt