diff --git a/actions/build-ml-whl/action.yaml b/actions/build-ml-whl/action.yaml
index 296a769..340943b 100644
--- a/actions/build-ml-whl/action.yaml
+++ b/actions/build-ml-whl/action.yaml
@@ -40,7 +40,7 @@ runs:
         if [[ ! -z "${{ inputs.venv }}" ]]; then
             source ${{ inputs.venv }}/bin/activate
         fi
-        pip install wheel
+        pip install setuptools wheel
         name="${{ inputs.name }}"
         ver_file=$(find . -type f -name "version.py")
         if ${{ inputs.release }}; then