diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml
index 89982b5d..a05261f0 100644
--- a/.github/workflows/publish.yml
+++ b/.github/workflows/publish.yml
@@ -27,10 +27,10 @@ jobs:
         run: python setup.py sdist bdist_wheel
 
       - name: Upload package artifact
-        uses: actions/upload-artifact@v2
+        uses: actions/upload-artifact@v4
         with:
           name: package
-          path: dist/*
+          path: dist
 
   test-package:
     needs: build-package
@@ -52,14 +52,15 @@ jobs:
           python-version: ${{ matrix.python-version }}
 
       - name: Download package artifact
-        uses: actions/download-artifact@v2
+        uses: actions/download-artifact@v4
         with:
           name: package
 
       - name: Install package and dependencies
         run: |
           python -m pip install --upgrade pip
-          pip install dist/*.whl
+          pip install plams-*.whl
+          pip install ".[chem_tools,results]"
           pip install pytest
 
       - name: Run Unit Tests
@@ -80,7 +81,7 @@ jobs:
           python-version: '3.x'
 
       - name: Download package artifact
-        uses: actions/download-artifact@v2
+        uses: actions/download-artifact@v4
         with:
           name: package