diff --git a/src/plugins/TreeViewPlugin.tsx b/src/plugins/TreeViewPlugin.tsx
index dd6d51d..31c41d4 100644
--- a/src/plugins/TreeViewPlugin.tsx
+++ b/src/plugins/TreeViewPlugin.tsx
@@ -19,6 +19,7 @@ export default function TreeViewPlugin(): JSX.Element {
       timeTravelButtonClassName="debug-timetravel-button"
       timeTravelPanelSliderClassName="debug-timetravel-panel-slider"
       timeTravelPanelButtonClassName="debug-timetravel-panel-button"
+      treeTypeButtonClassName="debug-treetype-button"
       editor={editor}
     />
   );