diff --git a/resources/views/layouts/tree-table.blade.php b/resources/views/layouts/tree-table.blade.php
index 9bd5fef..ad2ecd8 100644
--- a/resources/views/layouts/tree-table.blade.php
+++ b/resources/views/layouts/tree-table.blade.php
@@ -9,7 +9,7 @@
 @endempty
 
 <div class="bg-white rounded shadow-sm mb-3"
-     data-controller="table-tree"
+     data-controller="table-tree table"
      data-table-slug="{{$slug}}"
 >
 
diff --git a/src/Layouts/TDTree.php b/src/Layouts/TDTree.php
index 35009b4..8f17178 100644
--- a/src/Layouts/TDTree.php
+++ b/src/Layouts/TDTree.php
@@ -29,6 +29,8 @@ public function buildTd($repository, ?object $loop = null, bool $isFirstColumn =
             'slug' => $this->sluggable(),
             'width' => $this->width,
             'colspan' => $this->colspan,
+            'style'   => $this->style,
+            'class'   => $this->class,
         ]);
     }
 }