diff --git a/docs/ja/visualization.md b/docs/ja/visualization.md
index 3c5c64ec..9093bb65 100644
--- a/docs/ja/visualization.md
+++ b/docs/ja/visualization.md
@@ -81,7 +81,7 @@ draw_graph(triage_agent).view()
 デフォルトでは、`draw_graph` はグラフをインラインで表示します。ファイルとして保存するには、ファイル名を指定します:
 
 ```python
-draw_graph(triage_agent, filename="agent_graph.png")
+draw_graph(triage_agent, filename="agent_graph")
 ```
 
-これにより、作業ディレクトリに `agent_graph.png` が生成されます。
\ No newline at end of file
+これにより、作業ディレクトリに `agent_graph.png` が生成されます。
diff --git a/docs/visualization.md b/docs/visualization.md
index 00f3126d..409803f7 100644
--- a/docs/visualization.md
+++ b/docs/visualization.md
@@ -78,9 +78,7 @@ draw_graph(triage_agent).view()
 By default, `draw_graph` displays the graph inline. To save it as a file, specify a filename:
 
 ```python
-draw_graph(triage_agent, filename="agent_graph.png")
+draw_graph(triage_agent, filename="agent_graph")
 ```
 
 This will generate `agent_graph.png` in the working directory.
-
-
diff --git a/src/agents/extensions/visualization.py b/src/agents/extensions/visualization.py
index 5fb35062..888e262c 100644
--- a/src/agents/extensions/visualization.py
+++ b/src/agents/extensions/visualization.py
@@ -132,6 +132,6 @@ def draw_graph(agent: Agent, filename: Optional[str] = None) -> graphviz.Source:
     graph = graphviz.Source(dot_code)
 
     if filename:
-        graph.render(filename, format="png")
+        graph.render(filename, format="png", cleanup=True)
 
     return graph