diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
index 684b3195..408fae7b 100644
--- a/.github/workflows/main.yml
+++ b/.github/workflows/main.yml
@@ -6,23 +6,16 @@ jobs:
   test:
     name: Test
     runs-on: ubuntu-latest
-    strategy:
-      matrix:
-        container:
-          [
-            "ghcr.io/scientificcomputing/fenics-gmsh:2023-08-16",
-            "finsberg/pyadjoint-extra",
-          ]
 
     container:
-      image: ${{ matrix.container }}
+      image: "ghcr.io/scientificcomputing/fenics-gmsh:2024-02-19"
 
     steps:
-      - uses: actions/checkout@v3
+      - uses: actions/checkout@v4
 
       - name: Cache
         id: cache-primes
-        uses: actions/cache@v3
+        uses: actions/cache@v4
         with:
           path: |
             ~/.cache/instant
@@ -40,9 +33,10 @@ jobs:
 
 
       - name: Test with pytest
-        uses: nick-invision/retry@v2
-        with:
-          timeout_minutes: 120
-          max_attempts: 5
-          retry_on: error
-          command: python3 -m pytest
+        run: python3 -m pytest
+
+      - name: Install dolfin-adjoint
+        run:  python3 -m pip install dolfin-adjoint
+
+      - name: Test with pyadjoint
+        run: python3 -m pytest