diff --git a/.github/workflows/CI.yml b/.github/workflows/CI.yml
index 039747a0d..405f4cd14 100644
--- a/.github/workflows/CI.yml
+++ b/.github/workflows/CI.yml
@@ -74,7 +74,7 @@ jobs:
 
       - name: "Setup CI environment 🛠"
         # Important: make sure to update the SHA after making any changes to the set-dev-env action
-        uses: pydata/pydata-sphinx-theme/.github/actions/set-dev-env@01731d0cc57768b9eff1c97f38909932ecd7e7d1
+        uses: pydata/pydata-sphinx-theme/.github/actions/set-dev-env@4a1e7898d6c92dade5e489684277ab4ffd0eb053
         with:
           python-version: ${{ matrix.python-version }}
           pandoc: true
@@ -118,7 +118,7 @@ jobs:
           persist-credentials: false
 
       - name: "Setup CI environment 🛠"
-        uses: pydata/pydata-sphinx-theme/.github/actions/set-dev-env@01731d0cc57768b9eff1c97f38909932ecd7e7d1
+        uses: pydata/pydata-sphinx-theme/.github/actions/set-dev-env@4a1e7898d6c92dade5e489684277ab4ffd0eb053
         with:
           python-version: ${{ env.DEFAULT_PYTHON_VERSION }}
 
@@ -174,7 +174,7 @@ jobs:
           persist-credentials: false
 
       - name: "Setup CI environment 🛠"
-        uses: pydata/pydata-sphinx-theme/.github/actions/set-dev-env@01731d0cc57768b9eff1c97f38909932ecd7e7d1
+        uses: pydata/pydata-sphinx-theme/.github/actions/set-dev-env@4a1e7898d6c92dade5e489684277ab4ffd0eb053
         with:
           # 3.12 is not supported by py-spy yet
           python-version: "3.11"
diff --git a/.github/workflows/a11y.yml b/.github/workflows/a11y.yml
index e6445c0f9..0cb24fe07 100644
--- a/.github/workflows/a11y.yml
+++ b/.github/workflows/a11y.yml
@@ -46,7 +46,7 @@ jobs:
           persist-credentials: false
 
       - name: "Setup CI environment 🛠"
-        uses: pydata/pydata-sphinx-theme/.github/actions/set-dev-env@01731d0cc57768b9eff1c97f38909932ecd7e7d1
+        uses: pydata/pydata-sphinx-theme/.github/actions/set-dev-env@4a1e7898d6c92dade5e489684277ab4ffd0eb053
         with:
           python-version: ${{ env.DEFAULT_PYTHON_VERSION }}
           pandoc: true
diff --git a/.github/workflows/docs.yml b/.github/workflows/docs.yml
index 80742da91..b12ec341a 100644
--- a/.github/workflows/docs.yml
+++ b/.github/workflows/docs.yml
@@ -59,7 +59,7 @@ jobs:
 
       - name: "Setup CI environment 🛠"
         # Important: make sure to update the SHA after making any changes to the set-dev-env action
-        uses: pydata/pydata-sphinx-theme/.github/actions/set-dev-env@01731d0cc57768b9eff1c97f38909932ecd7e7d1
+        uses: pydata/pydata-sphinx-theme/.github/actions/set-dev-env@4a1e7898d6c92dade5e489684277ab4ffd0eb053
         with:
           python-version: ${{ matrix.python-version }}
           pandoc: true
@@ -90,7 +90,7 @@ jobs:
 
       - name: "Setup CI environment 🛠"
         # Important: make sure to update the SHA after making any changes to the set-dev-env action
-        uses: pydata/pydata-sphinx-theme/.github/actions/set-dev-env@01731d0cc57768b9eff1c97f38909932ecd7e7d1
+        uses: pydata/pydata-sphinx-theme/.github/actions/set-dev-env@4a1e7898d6c92dade5e489684277ab4ffd0eb053
         with:
           python-version: ${{ env.DEFAULT_PYTHON_VERSION }}
 
@@ -128,7 +128,7 @@ jobs:
 
       - name: "Setup CI environment 🛠"
         # Important: make sure to update the SHA after making any changes to the set-dev-env action
-        uses: pydata/pydata-sphinx-theme/.github/actions/set-dev-env@01731d0cc57768b9eff1c97f38909932ecd7e7d1
+        uses: pydata/pydata-sphinx-theme/.github/actions/set-dev-env@4a1e7898d6c92dade5e489684277ab4ffd0eb053
 
       - name: "Check for broken links 🔗"
         run: python -Im tox run -e docs-linkcheck
diff --git a/.github/workflows/prerelease.yml b/.github/workflows/prerelease.yml
index c78cdf40f..dbf429a98 100644
--- a/.github/workflows/prerelease.yml
+++ b/.github/workflows/prerelease.yml
@@ -31,7 +31,7 @@ jobs:
 
       - name: "Setup CI environment 🛠"
         # Important: make sure to update the SHA after making any changes to the set-dev-env action
-        uses: pydata/pydata-sphinx-theme/.github/actions/set-dev-env@01731d0cc57768b9eff1c97f38909932ecd7e7d1
+        uses: pydata/pydata-sphinx-theme/.github/actions/set-dev-env@4a1e7898d6c92dade5e489684277ab4ffd0eb053
         with:
           python-version: ${{ matrix.python-version }}
           pandoc: true
@@ -61,22 +61,16 @@ jobs:
         env:
           BAIPP_DIST: ${{ steps.baipp.outputs.dist }}
 
-      - name: "Get PST version"
-        run: |
-          python -Im pip install -e .
-          # Get the version of the package
-          PST_VERSION=$(python -c "import pydata_sphinx_theme; print(pydata_sphinx_theme.__version__)")
-          echo "PST_VERSION=$PST_VERSION" >> $GITHUB_ENV
-
       # Run tests on the built package (which will be later uploaded to PyPI)
       - name: "Install PST from wheel and test"
+        if: matrix.python-version == '3.9'
         env:
           BAIPP_DIST: ${{ steps.baipp.outputs.dist }}
-          PST_VERSION: ${{ env.PST_VERSION }}
         run: |
+          dist_loc="${BAIPP_DIST}"
           # calling with --installpkg so we can use the already built package
           tox run -e py312-tests-no-cov \
-          --installpkg "${BAIPP_DIST}"/pydata_sphinx_theme-"${PST_VERSION}"-py3-none-any.whl \
+          --installpkg $dist_loc/*.whl \
           -- --deselect tests/test_build.py::test_translations
 
       # If either the docs build or the tests resulted in an error, create an issue to note it
diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml
index 79eae1172..801ee2511 100644
--- a/.github/workflows/publish.yml
+++ b/.github/workflows/publish.yml
@@ -18,18 +18,16 @@ permissions:
 jobs:
   # calls our general CI workflows (tests, coverage, profile, etc.)
   tests:
-    # TODO @trallard: replace SHA after merge to main
     # Important: make sure to update the SHA after making any changes to the CI workflow
-    uses: pydata/pydata-sphinx-theme/.github/workflows/CI.yml@417d83135293eca5a3bbec100fece34467ced065
+    uses: pydata/pydata-sphinx-theme/.github/workflows/CI.yml@4a1e7898d6c92dade5e489684277ab4ffd0eb053
     # needed for the coverage action
     permissions:
       contents: write
       pull-requests: write
       # calls our docs workflow (build docs, check broken links, lighthouse)
   docs:
-    # TODO @trallard: replace SHA after merge to main
     # Important: make sure to update the SHA after making any changes to the docs workflow
-    uses: pydata/pydata-sphinx-theme/.github/workflows/docs.yml@417d83135293eca5a3bbec100fece34467ced065
+    uses: pydata/pydata-sphinx-theme/.github/workflows/docs.yml@4a1e7898d6c92dade5e489684277ab4ffd0eb053
 
   build-package:
     name: "Build & verify PST package"
@@ -57,6 +55,17 @@ jobs:
         env:
           BAIPP_DIST: ${{ steps.baipp.outputs.dist }}
 
+      # Run tests on the built package (which will be later uploaded to PyPI)
+      - name: "Install PST from wheel and test"
+        env:
+          BAIPP_DIST: ${{ steps.baipp.outputs.dist }}
+        run: |
+          dist_loc="${BAIPP_DIST}"
+          # calling with --installpkg so we can use the already built package
+          tox run -e py312-tests-no-cov \
+          --installpkg $dist_loc/*.whl \
+          -- --deselect tests/test_build.py::test_translations
+
   release-PST:
     runs-on: ubuntu-latest
     needs: [build-package]