diff --git a/.github/workflows/unit-tests.yml b/.github/workflows/unit-tests.yml
index 9fb1cd13db..2ac51cc3a2 100644
--- a/.github/workflows/unit-tests.yml
+++ b/.github/workflows/unit-tests.yml
@@ -12,7 +12,7 @@ jobs:
   run:
     strategy:
       matrix:
-        os: [ubuntu-24.04]
+        os: [ubuntu-latest]
         python-version: ["3.12", "3.11", "3.10"]
 
     runs-on: ${{ matrix.os }}
@@ -23,7 +23,7 @@ jobs:
       - name: Build River
         uses: ./.github/actions/install-env
         with:
-          python-version: "3.12"
+          python-version: ${{ matrix.python-version }}
 
       - name: Cache River datasets
         uses: actions/cache@v3
@@ -37,6 +37,9 @@ jobs:
           path: ~/scikit_learn_data
           key: ${{ runner.os }}
 
+      - name: Update Rust
+        run: rustup update
+
       - name: Download datasets
         run: |
           poetry run python -c "from river import datasets; datasets.CreditCard().download(); datasets.Elec2().download(); datasets.SMSSpam().download()"