diff --git a/docs/ai/quickstarts/snippets/image-generation/azure-openai/Program.cs b/docs/ai/quickstarts/snippets/image-generation/azure-openai/Program.cs
index 3ded4e9558e71..14eba55b69166 100644
--- a/docs/ai/quickstarts/snippets/image-generation/azure-openai/Program.cs
+++ b/docs/ai/quickstarts/snippets/image-generation/azure-openai/Program.cs
@@ -21,6 +21,13 @@
     A postal card with an happy hiker waving and a beautiful mountain in the background.
     There is a trail visible in the foreground.
     The postal card has text in red saying: 'You are invited for a hike!'
-    """, new ImageGenerationOptions { Size = GeneratedImageSize.W1024xH1024 });
+    """, new ImageGenerationOptions { Size = GeneratedImageSize.W1024xH1024, ResponseFormat = GeneratedImageFormat.Uri });
 
-Console.WriteLine($"The generated image is ready at:\n{generatedImage.ImageUri}");
+if (generatedImage.ImageUri != null)
+{
+    Console.WriteLine($"The generated image is ready at:\n{generatedImage.ImageUri}");
+}
+else
+{
+    Console.WriteLine("Error: Image URI is null. Check your API key and model name.");
+}