diff --git a/src/tree-node.svelte b/src/tree-node.svelte
index 43360e4..11cedd9 100644
--- a/src/tree-node.svelte
+++ b/src/tree-node.svelte
@@ -45,7 +45,7 @@
     {#if node.children}
       <div class="button" class:active={showChildren}>
         <svg focusable="false" viewBox="0 0 24 24">
-          <path d="M16.59 8.59L12 13.17 7.41 8.59 6 10l6 6 6-6z" />
+          <path d="M16.59 8.59L12 13.17 7.41 8.59 6 10l6 6 6-6z" class={$$props.class} />
         </svg>
       </div>
     {:else}
@@ -56,7 +56,7 @@
   {#if node.children && showChildren}
     <div class="children" transition:slide|local>
       {#each node.children as _node, i}
-        <svelte:self node={_node} index={i} let:node>
+        <svelte:self node={_node} index={i} let:node  class={$$props.class}>
           <slot {node} />
         </svelte:self>
       {/each}
diff --git a/src/tree.svelte b/src/tree.svelte
index 5b91341..641647b 100644
--- a/src/tree.svelte
+++ b/src/tree.svelte
@@ -4,7 +4,7 @@
 </script>
 
 {#each tree as _node, i}
-  <TreeNode node={_node} index={i} let:node>
+  <TreeNode node={_node} index={i} let:node class={$$props.class}>
     <slot {node}>
       <div class="name">{node.name}</div>
     </slot>