diff --git a/docs/ai/quickstarts/snippets/text-to-image/hosting/Program.cs b/docs/ai/quickstarts/snippets/text-to-image/hosting/Program.cs
index 3f23739b7127e..557a6c707003e 100644
--- a/docs/ai/quickstarts/snippets/text-to-image/hosting/Program.cs
+++ b/docs/ai/quickstarts/snippets/text-to-image/hosting/Program.cs
@@ -37,7 +37,7 @@
     ImageGenerationResponse response = await generator.GenerateImagesAsync(prompt);
     DataContent dataContent = response.Contents.OfType<DataContent>().First();
 
-    return Results.File(dataContent.Data.ToArray(), "image/png");
+    return Results.File(dataContent.Data.ToArray(), dataContent.MediaType);
 });
 // </SnippetUseImageGenerator>