diff --git a/addon/components/x-tree-node.js b/addon/components/x-tree-node.js
index 5d2dfaf..7a8e668 100644
--- a/addon/components/x-tree-node.js
+++ b/addon/components/x-tree-node.js
@@ -5,8 +5,8 @@ import layout from '../templates/components/x-tree-node';
 export default Component.extend({
   layout,
   classNameBindings: [
-    'model.isSelected:x-tree-highlight',
-    'isChosen:x-tree-chosen',
+    'model.isSelected:tree-highlight',
+    'isChosen:tree-chosen',
     'model.children.length:tree-children'
   ],