diff --git a/.github/workflows/task_runner_docker_e2e.yml b/.github/workflows/task_runner_docker_e2e.yml
index 5c31c7fbec..49942f4454 100644
--- a/.github/workflows/task_runner_docker_e2e.yml
+++ b/.github/workflows/task_runner_docker_e2e.yml
@@ -30,7 +30,9 @@ env:
 jobs:
   test_with_tls_docker:
     name: tr_tls_docker
-    runs-on: ubuntu-22.04
+    runs-on:
+      group: ubuntu-runners
+      labels: ubuntu-20.04-16core
     timeout-minutes: 15
     strategy:
       matrix:
@@ -66,6 +68,13 @@ jobs:
           pip install .
           pip install -r test-requirements.txt
 
+      - name: Create OpenFL image
+        id: create_openfl_image
+        run: |
+          echo "Creating openfl image. This may take a few minutes..."
+          cd openfl-docker
+          docker build . -t openfl -f Dockerfile.base
+
       - name: Run Task Runner E2E tests with TLS
         id: run_tests
         run: |
@@ -98,7 +107,9 @@ jobs:
 
   test_with_non_tls_docker:
     name: tr_non_tls_docker
-    runs-on: ubuntu-22.04
+    runs-on:
+      group: ubuntu-runners
+      labels: ubuntu-20.04-16core
     timeout-minutes: 15
     strategy:
       matrix:
@@ -134,6 +145,13 @@ jobs:
           pip install .
           pip install -r test-requirements.txt
 
+      - name: Create OpenFL image
+        id: create_openfl_image
+        run: |
+          echo "Creating openfl image. This may take a few minutes..."
+          cd openfl-docker
+          docker build . -t openfl -f Dockerfile.base
+
       - name: Run Task Runner E2E tests without TLS
         id: run_tests
         run: |
@@ -166,7 +184,9 @@ jobs:
 
   test_with_no_client_auth_docker:
     name: tr_no_client_auth_docker
-    runs-on: ubuntu-22.04
+    runs-on:
+      group: ubuntu-runners
+      labels: ubuntu-20.04-16core
     timeout-minutes: 15
     strategy:
       matrix:
@@ -202,6 +222,13 @@ jobs:
           pip install .
           pip install -r test-requirements.txt
 
+      - name: Create OpenFL image
+        id: create_openfl_image
+        run: |
+          echo "Creating openfl image. This may take a few minutes..."
+          cd openfl-docker
+          docker build . -t openfl -f Dockerfile.base
+
       - name: Run Task Runner E2E tests without TLS
         id: run_tests
         run: |
diff --git a/tests/end_to_end/utils/summary_helper.py b/tests/end_to_end/utils/summary_helper.py
index 8fcf45cebc..d71cdd15f4 100644
--- a/tests/end_to_end/utils/summary_helper.py
+++ b/tests/end_to_end/utils/summary_helper.py
@@ -11,7 +11,12 @@
 parser = etree.XMLParser(recover=True, encoding='utf-8')
 
 result_path = os.path.join(os.getenv("HOME"), "results")
-tree = ET.parse(f"{result_path}/results.xml", parser=parser)
+result_xml = os.path.join(result_path, "results.xml")
+if not os.path.exists(result_xml):
+    print(f"Results XML file not found at {result_xml}. Exiting...")
+    exit(1)
+
+tree = ET.parse(result_xml, parser=parser)
 
 # Get the root element
 testsuites = tree.getroot()