diff --git a/src/Umbraco.Web.UI.Client/src/common/directives/components/tree/umbtree.directive.js b/src/Umbraco.Web.UI.Client/src/common/directives/components/tree/umbtree.directive.js
index 8e38420886..07b3158a2a 100644
--- a/src/Umbraco.Web.UI.Client/src/common/directives/components/tree/umbtree.directive.js
+++ b/src/Umbraco.Web.UI.Client/src/common/directives/components/tree/umbtree.directive.js
@@ -28,8 +28,6 @@ function umbTreeDirective($compile, $log, $q, $rootScope, treeService, notificat
         },
         controller: function ($scope, $element) {
 
-            $scope.hideoptions = ($scope.hideoptions === 'true') ? "hide-options" : "";
-
             var vm = this;
 
             var registeredCallbacks = {
diff --git a/src/Umbraco.Web.UI.Client/src/views/components/tree/umb-tree.html b/src/Umbraco.Web.UI.Client/src/views/components/tree/umb-tree.html
index f26c2c1352..d30d9fb83a 100644
--- a/src/Umbraco.Web.UI.Client/src/views/components/tree/umb-tree.html
+++ b/src/Umbraco.Web.UI.Client/src/views/components/tree/umb-tree.html
@@ -1,4 +1,4 @@
-<ul class="umb-tree ' + hideoptions + '">
+<ul class="umb-tree" ng-class="{'hide-options': hideoptions === 'true'}">
     <li class="root">
         <div data-element="tree-root" ng-class="getNodeCssClass(tree.root)" ng-hide="hideheader" on-right-click="altSelect(tree.root, $event)">
             <h5>