diff --git a/src/agents/extensions/visualization.py b/src/agents/extensions/visualization.py
index 013a21e..9d0d9f6 100644
--- a/src/agents/extensions/visualization.py
+++ b/src/agents/extensions/visualization.py
@@ -111,7 +111,7 @@ def get_all_edges(agent: Agent, parent: Optional[Agent] = None) -> str:
             "{agent.name}" -> "{handoff.name}";""")
             parts.append(get_all_edges(handoff, agent))
 
-    if not agent.handoffs and not isinstance(agent, Tool):
+    if not agent.handoffs and not isinstance(agent, Tool): # type: ignore
         parts.append(f'"{agent.name}" -> "__end__";')
 
     return "".join(parts)