diff --git a/.github/workflows/test_onnxruntime_slow.yml b/.github/workflows/test_onnxruntime_slow.yml
index 92e2d4a8fc..f18881c1bb 100644
--- a/.github/workflows/test_onnxruntime_slow.yml
+++ b/.github/workflows/test_onnxruntime_slow.yml
@@ -27,7 +27,8 @@ jobs:
       contains(github.event.pull_request.labels.*.name, 'onnxruntime-slow')
       }}
 
-    runs-on: ubuntu-20.04
+    runs-on:
+      group: aws-general-8-plus
 
     steps:
       - name: Checkout
diff --git a/tests/onnxruntime/test_modeling.py b/tests/onnxruntime/test_modeling.py
index 9dbcb1faca..61c78bb378 100644
--- a/tests/onnxruntime/test_modeling.py
+++ b/tests/onnxruntime/test_modeling.py
@@ -1042,7 +1042,7 @@ def test_save_load_ort_model_with_external_data(self):
     def test_save_load_decoder_model_with_external_data(self, use_cache: bool):
         with tempfile.TemporaryDirectory() as tmpdirname:
             model = ORTModelForCausalLM.from_pretrained(
-                "gpt2-large", use_cache=use_cache, export=True, use_merged=False, use_io_binding=False
+                "gpt2-large", export=True, use_cache=use_cache, use_merged=False, use_io_binding=False
             )
             model.save_pretrained(tmpdirname)