diff --git a/test/oai/test_completion.py b/test/oai/test_completion.py
index 934dbadc9016..f33533b7bf88 100644
--- a/test/oai/test_completion.py
+++ b/test/oai/test_completion.py
@@ -376,7 +376,7 @@ def test_math(num_samples=-1):
     ]
 
     autogen.Completion.set_cache(seed)
-    config_list = autogen.config_list_openai_aoai(KEY_LOC, exclude="aoai")
+    config_list = autogen.config_list_openai_aoai(KEY_LOC)[:2]
     vanilla_config = {
         "model": "text-davinci-003",
         "temperature": 1,