diff --git a/lua/u/renderer.lua b/lua/u/renderer.lua
index cd11bc2..f4030b0 100644
--- a/lua/u/renderer.lua
+++ b/lua/u/renderer.lua
@@ -386,23 +386,23 @@ function Renderer:get_pos_infos(pos0)
 end
 
 --------------------------------------------------------------------------------
--- TagWriter class
+-- TreeBuilder class
 --------------------------------------------------------------------------------
 
---- @class TagWriter
+--- @class TreeBuilder
 --- @field private nodes Node[]
-local TagWriter = {}
-TagWriter.__index = TagWriter
-M.TagWriter = TagWriter
+local TreeBuilder = {}
+TreeBuilder.__index = TreeBuilder
+M.TreeBuilder = TreeBuilder
 
-function TagWriter.new()
-  local self = setmetatable({ nodes = {} }, TagWriter)
+function TreeBuilder.new()
+  local self = setmetatable({ nodes = {} }, TreeBuilder)
   return self
 end
 
 --- @param nodes Tree
---- @return TagWriter
-function TagWriter:put(nodes)
+--- @return TreeBuilder
+function TreeBuilder:put(nodes)
   table.insert(self.nodes, nodes)
   return self
 end
@@ -410,23 +410,23 @@ end
 --- @param name string
 --- @param attributes? table<string, any>
 --- @param children? Node | Node[]
---- @return TagWriter
-function TagWriter:put_h(name, attributes, children)
+--- @return TreeBuilder
+function TreeBuilder:put_h(name, attributes, children)
   local tag = M.h(name, attributes, children)
   table.insert(self.nodes, tag)
   return self
 end
 
---- @param fn fun(TagWriter): any
---- @return TagWriter
-function TagWriter:nest(fn)
-  local nested_writer = TagWriter.new()
+--- @param fn fun(TreeBuilder): any
+--- @return TreeBuilder
+function TreeBuilder:nest(fn)
+  local nested_writer = TreeBuilder.new()
   fn(nested_writer)
   table.insert(self.nodes, nested_writer.nodes)
   return self
 end
 
 --- @return Tree
-function TagWriter:tree() return self.nodes end
+function TreeBuilder:tree() return self.nodes end
 
 return M