diff --git a/.github/workflows/python-package.yml b/.github/workflows/python-package.yml
index 5704b59e..6e3c980d 100644
--- a/.github/workflows/python-package.yml
+++ b/.github/workflows/python-package.yml
@@ -4,14 +4,18 @@ on: [ push, pull_request ]
 
 jobs:
   test:
-    # TODO: a GH action update broke the 'ubuntu-latest' image
-    #       when it's fixed, we should switch back
-    runs-on: ubuntu-20.04
+    runs-on: ${{ matrix.os }}
 
     strategy:
       fail-fast: false
       matrix:
-        python-version: ['3.5', '3.6', '3.7', '3.8', '3.9', '3.10', '3.11']
+        python-version: ['3.7', '3.8', '3.9', '3.10', '3.11']
+        os: ['ubuntu-latest']
+        include:
+          - python-version: '3.5'
+            os: 'ubuntu-20.04'
+          - python-version: '3.6'
+            os: 'ubuntu-20.04'
 
     steps:
     - uses: actions/checkout@v3