diff --git a/.github/workflows/UnitTests.yml b/.github/workflows/UnitTests.yml
index f7ee2166..44fb5bf1 100644
--- a/.github/workflows/UnitTests.yml
+++ b/.github/workflows/UnitTests.yml
@@ -31,7 +31,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        tpu-type: ["v4-8"]
+        tpu-type: ["v5-8"]
     name: "TPU test (${{ matrix.tpu-type }})"
     runs-on: ["self-hosted", "tpu", "${{ matrix.tpu-type }}"]
     steps:
@@ -48,7 +48,7 @@ jobs:
         export PATH=$PATH:$HOME/.local/bin
         pip install ruff
         pip install isort
-        pip install pytest
+        pip install pytest==8.2.2
     - name: Analysing the code with ruff
       run: |
         ruff check .