diff --git a/addon/templates/components/x-tree-node.hbs b/addon/templates/components/x-tree-node.hbs
index 381f964..ce12440 100644
--- a/addon/templates/components/x-tree-node.hbs
+++ b/addon/templates/components/x-tree-node.hbs
@@ -12,6 +12,8 @@
         model=model
         toggleCheck=(action "toggleCheck")
       )
+      toggleExpand=(action "toggleExpand")
+      toggleCheck=(action "toggleCheck")
     )
   }}
 {{else}}