diff --git a/.github/workflows/jedi.yml b/.github/workflows/jedi.yml
index c5305f5..95bef3c 100644
--- a/.github/workflows/jedi.yml
+++ b/.github/workflows/jedi.yml
@@ -7,10 +7,7 @@ jobs:
     if: github.ref_type != 'tag'
     runs-on: ubuntu-22.04
     steps:
-    - uses: actions/checkout@v2
-    - uses: actions/setup-python@v2
-      with:
-        python-version: "3.10"
+    - uses: actions/checkout@v3
     - run: pipx install poetry
     - run: poetry install
       working-directory: ./jedi
@@ -19,12 +16,9 @@ jobs:
 
   publish:
     if: github.ref_type == 'tag' && startsWith(github.ref_name, 'pybricks_jedi/')
-    runs-on: ubuntu-20.04
+    runs-on: ubuntu-22.04
     steps:
-    - uses: actions/checkout@v2
-    - uses: actions/setup-python@v2
-      with:
-        python-version: "3.10"
+    - uses: actions/checkout@v3
     - run: pipx install poetry
     - run: poetry install
       working-directory: ./jedi