diff --git a/docs/visualization.md b/docs/visualization.md
index 6bb5ee5..2e3b188 100644
--- a/docs/visualization.md
+++ b/docs/visualization.md
@@ -7,7 +7,7 @@ Agent visualization allows you to generate a structured graphical representation
 The visualization functionality relies on the **Graphviz** package. To use it, ensure you have Graphviz installed and add it as a dependency in `pyproject.toml`. Alternatively, install it directly via pip:
 
 ```bash
-pip install openai-agents[visualization]
+pip install openai-agents[viz]
 ```
 
 ## Generating a Graph