diff --git a/docs/contributing/model/adding_multi_stage_model.md b/docs/contributing/model/adding_multi_stage_model.md
index 9f89b3b45..71b0c6c72 100644
--- a/docs/contributing/model/adding_multi_stage_model.md
+++ b/docs/contributing/model/adding_multi_stage_model.md
@@ -476,8 +476,8 @@ thinker_hidden_states = output.multimodal_output["24"]
 
 <p align="center">
   <picture>
-    <source media="(prefers-color-scheme: dark)" src="https://raw.githubusercontent.com/vllm-project/vllm-omni/refs/heads/main/docs/source/architecture/omni-modality-model-architecture.png">
-    <img alt="Data Flow between stages" src="image.png" width=55%>
+    <source media="(prefers-color-scheme: dark)" src="https://raw.githubusercontent.com/vllm-project/vllm-omni/refs/heads/main/docs/source/architecture/vllm-omni-dataflow-between-stages.png">
+    <img alt="Data Flow between stages" src="https://raw.githubusercontent.com/vllm-project/vllm-omni/refs/heads/main/docs/source/architecture/vllm-omni-dataflow-between-stages.png" width=55%>
   </picture>
 </p>