diff --git a/action.yaml b/action.yaml
index c7c9d93..c9a49c0 100644
--- a/action.yaml
+++ b/action.yaml
@@ -14,12 +14,14 @@ runs:
         else
           . $VENV/bin/activate
         fi
-        python -m pip install -U pip setuptools
+        python -m pip install -U pip setuptools wheel
         echo ::endgroup::
 
         echo ::group::Install dependencies
         # Remove this package from constraints
-        grep -v "^$(python setup.py --name)@" ${GITHUB_ACTION_PATH}/constraints.txt > constraints.txt
+        PKG_NAME=$(pip download . --no-deps --pre --no-build-isolation |  grep -i 'Successfully downloaded' | awk '{ print $NF }')
+        echo "Removing ${PKG_NAME} from constraints.txt"
+        grep -v "^${PKG_NAME}@" ${GITHUB_ACTION_PATH}/constraints.txt > constraints.txt
         # Install dependencies, including any 'test' extras, as well as pytest-cov
         python -m pip install -U -e .[test] pytest-cov -c constraints.txt
         echo ::endgroup::