diff --git a/.github/workflows/pr-cpu.yaml b/.github/workflows/pr-cpu.yaml
index 056b070143..618e64f7b7 100644
--- a/.github/workflows/pr-cpu.yaml
+++ b/.github/workflows/pr-cpu.yaml
@@ -16,7 +16,7 @@ concurrency:
 jobs:
   pytest-cpu:
     name: ${{ matrix.name }}
-    runs-on: ubuntu-latest
+    runs-on: linux-ubuntu-latest
     container: ${{ matrix.container }}
     strategy:
       matrix:
diff --git a/tests/a_scripts/train/test_train.py b/tests/a_scripts/train/test_train.py
index b1bca9ebd0..e752fb0d40 100644
--- a/tests/a_scripts/train/test_train.py
+++ b/tests/a_scripts/train/test_train.py
@@ -115,6 +115,8 @@ def test_train_gauntlet(averages: Optional[dict], tmp_path: pathlib.Path):
 
 
 def test_sort_callbacks():
+    import time
+    time.sleep(1000)
     trainer_mock = Mock()
     trainer_mock.state.callbacks = [
         CheckpointSaver(),