diff --git a/tests/image_gen_tests/test_image_generation.py b/tests/image_gen_tests/test_image_generation.py
index 63274dfedf3..bee9d1f7d50 100644
--- a/tests/image_gen_tests/test_image_generation.py
+++ b/tests/image_gen_tests/test_image_generation.py
@@ -168,6 +168,8 @@ class TestAzureOpenAIDalle3(BaseImageGenTest):
         return {
             "model": "azure/dall-e-3-test",
             "api_version": "2023-12-01-preview",
+            "api_base": os.getenv("AZURE_SWEDEN_API_BASE"),
+            "api_key": os.getenv("AZURE_SWEDEN_API_KEY"),
             "metadata": {
                 "model_info": {
                     "base_model": "azure/dall-e-3",