diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
index 3beed64..0b84147 100644
--- a/.github/workflows/build.yaml
+++ b/.github/workflows/build.yaml
@@ -162,7 +162,7 @@ jobs:
       #     user: __token__
       #     password: ${{ secrets.pypi_password }}
       - name: Store the binary wheel
-        uses: actions/upload-artifact@v2
+        uses: actions/upload-artifact@v4
         with:
           name: python-package-distributions
           path: dist
@@ -174,7 +174,7 @@ jobs:
 
     steps:
       - name: Download all the dists
-        uses: actions/download-artifact@v2
+        uses: actions/download-artifact@v4
         with:
           name: python-package-distributions
           path: dist/
diff --git a/tools/build-wheels-dev.sh b/tools/build-wheels-dev.sh
index 89a5b08..64a75e1 100755
--- a/tools/build-wheels-dev.sh
+++ b/tools/build-wheels-dev.sh
@@ -13,7 +13,7 @@ function repair_wheel {
 
 export CUMM_DISABLE_JIT="1"
 export CUMM_CUDA_ARCH_LIST="7.5;8.6"
-"/opt/python/cp39-cp39/bin/pip" wheel /io/ --no-deps -w /io/wheelhouse_tmp
+"/opt/python/cp313-cp313t/bin/pip" wheel /io/ --no-deps -w /io/wheelhouse_tmp
 
 # Bundle external shared libraries into the wheels
 for whl in /io/wheelhouse_tmp/*.whl; do