diff --git a/.circleci/config.yml b/.circleci/config.yml
index de6ba01531..b1e08084fa 100644
--- a/.circleci/config.yml
+++ b/.circleci/config.yml
@@ -997,6 +997,7 @@ jobs:
 
     steps:
       - checkout
+      - setup_google_dns
       - run:
           name: Install Dependencies
           command: |
diff --git a/tests/image_gen_tests/base_image_generation_test.py b/tests/image_gen_tests/base_image_generation_test.py
index 25529c9e0b..cf01390e07 100644
--- a/tests/image_gen_tests/base_image_generation_test.py
+++ b/tests/image_gen_tests/base_image_generation_test.py
@@ -44,7 +44,7 @@ class BaseImageGenTest(ABC):
         pass
 
     @pytest.mark.asyncio(scope="module")
-    async def test_async_basic_image_generation(self):
+    async def test_basic_image_generation(self):
         """Test basic image generation"""
         try:
             custom_logger = TestCustomLogger()