diff --git a/pipelines/templates/stages/build-python.yaml b/pipelines/templates/stages/build-python.yaml
index 0398a43..38e3690 100644
--- a/pipelines/templates/stages/build-python.yaml
+++ b/pipelines/templates/stages/build-python.yaml
@@ -38,35 +38,35 @@ stages:
               git submodule foreach git pull
               git submodule foreach git tag ${TAG}
               git submodule foreach git push origin ${TAG}
-              sudo env "PATH=$PATH" python -m pip install --no-cache-dir -U wheel cython pip
+              sudo env "PATH=$PATH" python -m pip install --no-cache-dir -U wheel cython pip build
 
               # Build base
               cd assemblyline-base
               echo $VERSION > assemblyline/VERSION
-              python setup.py sdist --dist-dir ${SYSTEM_DEFAULTWORKINGDIRECTORY}/dist/ &
+              python -m build -s -o ${SYSTEM_DEFAULTWORKINGDIRECTORY}/dist/ &
 
               # Build core
               cd ../assemblyline-core
               echo $VERSION > assemblyline_core/VERSION
-              python setup.py sdist --dist-dir ${SYSTEM_DEFAULTWORKINGDIRECTORY}/dist/ &
+              python -m build -s -o ${SYSTEM_DEFAULTWORKINGDIRECTORY}/dist/ &
 
               # Build ui
               cd ../assemblyline-ui
               echo $VERSION > assemblyline_ui/VERSION
-              python setup.py bdist_wheel --dist-dir ${SYSTEM_DEFAULTWORKINGDIRECTORY}/dist/ &
+              python -m build -w -o ${SYSTEM_DEFAULTWORKINGDIRECTORY}/dist/ &
 
               # Build service server
               cd ../assemblyline-service-server
               echo $VERSION > assemblyline_service_server/VERSION
-              python setup.py bdist_wheel --dist-dir ${SYSTEM_DEFAULTWORKINGDIRECTORY}/dist/ &
+              python -m build -w -o ${SYSTEM_DEFAULTWORKINGDIRECTORY}/dist/ &
 
               cd ../assemblyline-service-client
               echo $VERSION > assemblyline_service_client/VERSION
-              python setup.py bdist_wheel --dist-dir ${SYSTEM_DEFAULTWORKINGDIRECTORY}/dist/ &
+              python -m build -w -o ${SYSTEM_DEFAULTWORKINGDIRECTORY}/dist/ &
 
               cd ../assemblyline-v4-service
               echo $VERSION > assemblyline_v4_service/VERSION
-              python setup.py bdist_wheel --dist-dir ${SYSTEM_DEFAULTWORKINGDIRECTORY}/dist/ &
+              python -m build -w -o ${SYSTEM_DEFAULTWORKINGDIRECTORY}/dist/ &
 
               wait
               cd ../