diff --git a/.github/workflows/pypi_publish.yml b/.github/workflows/pypi_publish.yml
index 2c6cdef1..8a4e4c73 100644
--- a/.github/workflows/pypi_publish.yml
+++ b/.github/workflows/pypi_publish.yml
@@ -39,11 +39,15 @@ jobs:
           export MODE="omp"
           echo "MODE=${MODE}" >> $GITHUB_ENV
 
-      - name: Set up Visual Studio shell
+      # - name: Set up Visual Studio shell
+      #   if: runner.os == 'Windows'
+      #   uses: egor-tensin/vs-shell@v2
+      #   with:
+      #     arch: x64
+
+      - name: Add msbuild to PATH
         if: runner.os == 'Windows'
-        uses: egor-tensin/vs-shell@v2
-        with:
-          arch: x64
+        uses: microsoft/setup-msbuild@v2
 
       - name: Build wheels
         uses: pypa/cibuildwheel@v2.17.0