diff --git a/tests/test_image_gen.py b/tests/integration/test_image_gen.py
similarity index 97%
rename from tests/test_image_gen.py
rename to tests/integration/test_image_gen.py
index 5c04921b5a52..0156c9e5bdc9 100644
--- a/tests/test_image_gen.py
+++ b/tests/integration/test_image_gen.py
@@ -16,11 +16,9 @@ def image_size(request):
     return request.param
 
 
-@pytest.mark.xfail(
-    reason="The image is too big to be put in a cassette for a CI pipeline. We're looking into a solution."
-)
 @requires_api_key("OPENAI_API_KEY")
-def test_dalle(config, workspace, image_size):
+@pytest.mark.vcr
+def test_dalle(config, workspace, image_size, patched_api_requestor):
     """Test DALL-E image generation."""
     generate_and_validate(
         config,