diff --git a/lib/outlook_web/components/html_doc_component.ex b/lib/outlook_web/components/html_doc_component.ex
index 099837b..bca3694 100644
--- a/lib/outlook_web/components/html_doc_component.ex
+++ b/lib/outlook_web/components/html_doc_component.ex
@@ -8,17 +8,13 @@ defmodule OutlookWeb.HtmlDocComponent do
   alias Phoenix.LiveView.JS
 
   attr :tree, :list, required: true
+  attr :tunit_tag, :atom, default: :span
 
-  def render_doc(%{tunit_tag: _} = assigns) do
+  def render_doc(assigns) do
     ~H"""
     <.dnode :for={node <- @tree} node={node} tunit_tag={@tunit_tag} />
     """
   end
-  def render_doc(assigns) do
-    assigns
-    |> Map.put(:tunit_tag,  "span")
-    |> render_doc()
-  end
 
   def dnode(%{node: %{status: _}} = assigns) do
     ~H"""