diff --git a/tests/e2e-aio/e2e_test.go b/tests/e2e-aio/e2e_test.go
index 127626e2..c9271e12 100644
--- a/tests/e2e-aio/e2e_test.go
+++ b/tests/e2e-aio/e2e_test.go
@@ -124,7 +124,7 @@ var _ = Describe("E2E test", func() {
 				req := openai.ImageRequest{
 					Prompt:  "test",
 					Quality: "1",
-					Size:    openai.CreateImageSize512x512,
+					Size:    openai.CreateImageSize256x256,
 				}
 				resp, err := client.CreateImage(context.TODO(), req)
 				Expect(err).ToNot(HaveOccurred(), fmt.Sprintf("error sending image request %+v", req))