diff --git a/.github/workflows/publish_pypi.yml b/.github/workflows/publish_pypi.yml
index 11c4553ead..6160c7caaf 100644
--- a/.github/workflows/publish_pypi.yml
+++ b/.github/workflows/publish_pypi.yml
@@ -121,10 +121,9 @@ jobs:
             yum -y install epel-release &&
             yum -y install openblas-devel lapack-devel &&
             bash scripts/install_sundials.sh 6.0.3 6.5.0
-          CIBW_BEFORE_BUILD_LINUX: >
-            python -m pip install cmake casadi setuptools wheel
-          CIBW_REPAIR_WHEEL_COMMAND_LINUX: >
-            auditwheel repair -w {dest_dir} {wheel}
+          CIBW_BEFORE_BUILD_LINUX: python -m pip install cmake casadi setuptools wheel
+          CIBW_REPAIR_WHEEL_COMMAND_LINUX: auditwheel repair -w {dest_dir} {wheel}
+          CIBW_TEST_COMMAND: python -c "import pybamm; print(pybamm.have_idaklu())" | grep True
 
       - name: Build Linux wheels for ${{ matrix.arch }}
         run: pipx run cibuildwheel --output-dir wheelhouse
@@ -134,19 +133,19 @@ jobs:
           CIBW_BEFORE_ALL_LINUX: >
             yum -y install openblas-devel lapack-devel &&
             bash scripts/install_sundials.sh 6.0.3 6.5.0
-          CIBW_BEFORE_BUILD_LINUX: >
-            python -m pip install cmake casadi setuptools wheel
-          CIBW_REPAIR_WHEEL_COMMAND_LINUX: >
-            auditwheel repair -w {dest_dir} {wheel}
+          CIBW_BEFORE_BUILD_LINUX: python -m pip install cmake casadi setuptools wheel
+          CIBW_REPAIR_WHEEL_COMMAND_LINUX: auditwheel repair -w {dest_dir} {wheel}
+          CIBW_TEST_COMMAND: python -c "import pybamm; print(pybamm.have_idaklu())" | grep True
 
       - name: Build macOS wheels for ${{ matrix.arch }}
         if: matrix.os == 'macos-latest'
         run: pipx run cibuildwheel --output-dir wheelhouse
         env:
           CIBW_BEFORE_BUILD_MACOS: >
-            python -m pip install --upgrade cmake casadi setuptools wheel && scripts/fix_suitesparse_rpath_mac.sh
-          CIBW_REPAIR_WHEEL_COMMAND_MACOS: >
-            delocate-listdeps {wheel} && delocate-wheel -v -w {dest_dir} {wheel}
+            python -m pip install --upgrade cmake casadi setuptools wheel &&
+            scripts/fix_suitesparse_rpath_mac.sh
+          CIBW_REPAIR_WHEEL_COMMAND_MACOS: delocate-listdeps {wheel} && delocate-wheel -v -w {dest_dir} {wheel}
+          CIBW_TEST_COMMAND: python -c "import pybamm; print(pybamm.have_idaklu())" | grep True
 
       - name: Upload wheels
         uses: actions/upload-artifact@v3