diff --git a/cms/static/js/views/pages/container.js b/cms/static/js/views/pages/container.js
index 1e8e5e8b0c..7e1f6de3b6 100644
--- a/cms/static/js/views/pages/container.js
+++ b/cms/static/js/views/pages/container.js
@@ -367,7 +367,7 @@ function($, _, Backbone, gettext, BasePage,
event.preventDefault();
if (!options || options.view !== 'visibility_view') {
- const primaryHeader = $(event.target).closest('.xblock-header-primary');
+ const primaryHeader = $(event.target).closest('.xblock-header-primary, .nav-actions');
var useNewTextEditor = primaryHeader.attr('use-new-editor-text'),
useNewVideoEditor = primaryHeader.attr('use-new-editor-video'),
diff --git a/cms/templates/container.html b/cms/templates/container.html
index 488ad2a3b7..058587e6bb 100644
--- a/cms/templates/container.html
+++ b/cms/templates/container.html
@@ -13,6 +13,8 @@ from django.urls import reverse
from django.utils.translation import gettext as _
from cms.djangoapps.contentstore.helpers import xblock_studio_url, xblock_type_display_name
+from cms.djangoapps.contentstore.toggles import use_new_text_editor, use_new_problem_editor, use_new_video_editor, use_video_gallery_flow
+from cms.djangoapps.contentstore.utils import get_editor_page_base_url
from openedx.core.djangolib.js_utils import (
dump_js_escaped_json, js_escaped_string
)
@@ -111,6 +113,13 @@ from openedx.core.djangolib.markup import HTML, Text
<%block name="content">
+<%
+use_new_editor_text = use_new_text_editor()
+use_new_editor_video = use_new_video_editor()
+use_new_editor_problem = use_new_problem_editor()
+use_new_video_gallery_flow = use_video_gallery_flow()
+%>
+
@@ -161,7 +170,15 @@ from openedx.core.djangolib.markup import HTML, Text
-