diff --git a/azure-pipelines.yml b/azure-pipelines.yml
index 4be3f33af0b..ca74633cafa 100644
--- a/azure-pipelines.yml
+++ b/azure-pipelines.yml
@@ -51,7 +51,7 @@ jobs:
   steps:
   - script: "/opt/python/cp38-cp38/bin/python -m venv $PWD/env"
   - script: 'echo "##vso[task.prependpath]$PWD/env/bin"'
-  - script: "pip install build git+https://github.com/rhelmot/auditwheel"  # @TODO remove when patches make it upstream
+  - script: "pip install build"
   - script: "cd src/api/python && python -m build && AUDITWHEEL_PLAT= auditwheel repair --best-plat dist/*.whl && cd ../../.."
   - script: "pip install ./src/api/python/wheelhouse/*.whl && python - <src/api/python/z3test.py z3 && python - <src/api/python/z3test.py z3num"
 
@@ -73,6 +73,7 @@ jobs:
   - script: echo '##vso[task.prependpath]/tmp/arm-toolchain/aarch64-none-linux-gnu/libc/usr/bin'
   - script: echo $PATH
   - script: "stat `which aarch64-none-linux-gnu-gcc`"
+  - script: "pip install build"
   - script: "cd src/api/python && CC=aarch64-none-linux-gnu-gcc CXX=aarch64-none-linux-gnu-g++ AR=aarch64-none-linux-gnu-ar LD=aarch64-none-linux-gnu-ld python -m build && AUDITWHEEL_PLAT= auditwheel repair --best-plat dist/*.whl && cd ../../.."
 
 - job: "Ubuntu20OCaml"