diff --git a/docs/getting_started/quickstart.md b/docs/getting_started/quickstart.md
index 423123ac0..e602209c3 100644
--- a/docs/getting_started/quickstart.md
+++ b/docs/getting_started/quickstart.md
@@ -35,7 +35,8 @@ from vllm_omni.entrypoints.omni import Omni
 if __name__ == "__main__":
     omni = Omni(model="Tongyi-MAI/Z-Image-Turbo")
     prompt = "a cup of coffee on the table"
-    images = omni.generate(prompt)
+    outputs = omni.generate(prompt)
+    images = outputs[0].request_output[0]["images"]
     images[0].save("coffee.png")
 ```
 
diff --git a/docs/user_guide/examples/offline_inference/text_to_image.md b/docs/user_guide/examples/offline_inference/text_to_image.md
index a2e427b25..a89da2006 100644
--- a/docs/user_guide/examples/offline_inference/text_to_image.md
+++ b/docs/user_guide/examples/offline_inference/text_to_image.md
@@ -16,7 +16,8 @@ from vllm_omni.entrypoints.omni import Omni
 if __name__ == "__main__":
     omni = Omni(model="Qwen/Qwen-Image")
     prompt = "a cup of coffee on the table"
-    images = omni.generate(prompt)
+    outputs = omni.generate(prompt)
+    images = outputs[0].request_output[0]["images"]
     images[0].save("coffee.png")
 ```
 
diff --git a/examples/offline_inference/text_to_image/README.md b/examples/offline_inference/text_to_image/README.md
index d6291076a..b0c7baab5 100644
--- a/examples/offline_inference/text_to_image/README.md
+++ b/examples/offline_inference/text_to_image/README.md
@@ -13,7 +13,8 @@ from vllm_omni.entrypoints.omni import Omni
 if __name__ == "__main__":
     omni = Omni(model="Qwen/Qwen-Image")
     prompt = "a cup of coffee on the table"
-    images = omni.generate(prompt)
+    outputs = omni.generate(prompt)
+    images = outputs[0].request_output[0]["images"]
     images[0].save("coffee.png")
 ```