diff --git a/src/cleveragents/tui/_tui_renderers.py b/src/cleveragents/tui/_tui_renderers.py
index d1ad1f9d7..72416313d 100644
--- a/src/cleveragents/tui/_tui_renderers.py
+++ b/src/cleveragents/tui/_tui_renderers.py
@@ -18,6 +18,7 @@ from cleveragents.cli.output.handles._models import (
     StatusMessage,
     Table,
     TextBlock,
+    TreeNode,
     Tree,
 )
 
@@ -98,16 +99,16 @@ def _render_tree(element: Tree) -> str:
     """Render a Tree element as plain text."""
     lines: list[str] = []
 
-    def _render_node(node: object, prefix: str, is_last: bool) -> None:
+    def _render_node(node: TreeNode, prefix: str, is_last: bool) -> None:
         connector = "L-- " if is_last else "+-- "
-        lines.append(f"{prefix}{connector}{node.label}")  # type: ignore[union-attr]
+        lines.append(f"{prefix}{connector}{node.label}")
         child_prefix = prefix + ("    " if is_last else "|   ")
-        for i, child in enumerate(node.children):  # type: ignore[union-attr]
-            _render_node(child, child_prefix, i == len(node.children) - 1)  # type: ignore[union-attr]
+        for i, child in enumerate(node.children):
+            _render_node(child, child_prefix, i == len(node.children) - 1)
 
     root = element.root
     lines.append(root.label)
-    for i, child in enumerate(root.children):  # type: ignore[union-attr]
+    for i, child in enumerate(root.children):
         _render_node(child, "", i == len(root.children) - 1)
     return "\n".join(lines)