diff --git a/src/Umbraco.Web.UI.Client/src/views/components/editor/umb-editor-content-header.html b/src/Umbraco.Web.UI.Client/src/views/components/editor/umb-editor-content-header.html
index 149fccd00a..e0f40a1b3b 100644
--- a/src/Umbraco.Web.UI.Client/src/views/components/editor/umb-editor-content-header.html
+++ b/src/Umbraco.Web.UI.Client/src/views/components/editor/umb-editor-content-header.html
@@ -72,7 +72,7 @@
             <umb-editor-menu 
                 data-element="editor-actions"
                 current-node="menu.currentNode"
-                current-section="{{menu.currentSection}}">
+                current-section="{{menu.currentNode.section}}">
             </umb-editor-menu>
         </div>
 
diff --git a/src/Umbraco.Web.UI.Client/src/views/components/editor/umb-editor-menu.html b/src/Umbraco.Web.UI.Client/src/views/components/editor/umb-editor-menu.html
index f724f39be7..bf9c8fab8c 100644
--- a/src/Umbraco.Web.UI.Client/src/views/components/editor/umb-editor-menu.html
+++ b/src/Umbraco.Web.UI.Client/src/views/components/editor/umb-editor-menu.html
@@ -8,7 +8,7 @@
 
     <!-- actions -->
     <ul class="dropdown-menu umb-actions" role="menu" aria-labelledby="dLabel">
-        <li class="umb-action" ng-class="{sep:action.seperator}" ng-repeat="action in actions">
+        <li class="umb-action" ng-class="{'sep':action.seperatorm, '-opens-dialog': action.opensDialog}" ng-repeat="action in actions">
 
             <!-- How does this reference executeMenuItem() i really don't think that this is very clear -->
             <a prevent-default
@@ -17,7 +17,6 @@
                 <i class="icon icon-{{action.cssclass}}"></i>
                 <span class="menu-label">{{action.name}}</span>
             </a>
-
         </li>
     </ul>
 </div>