diff --git a/cms/templates/container.html b/cms/templates/container.html
index d6825fa1aa..4610acfa48 100644
--- a/cms/templates/container.html
+++ b/cms/templates/container.html
@@ -87,12 +87,12 @@ templates = ["basic-modal", "modal-button", "edit-xblock-modal",
% if is_unit_page:
-
-
+
${_("View Live Version")}
-
-
+
${_("Preview Changes")}
diff --git a/common/test/acceptance/tests/test_studio_container.py b/common/test/acceptance/tests/test_studio_container.py
index 73c65db501..0646ea5e13 100644
--- a/common/test/acceptance/tests/test_studio_container.py
+++ b/common/test/acceptance/tests/test_studio_container.py
@@ -692,6 +692,7 @@ class UnitPublishingTest(ContainerBase):
Goes to the published version, then waits for the browser to load the page.
"""
unit.view_published_version()
+ self.assertEqual(len(self.browser.window_handles), 2)
self.courseware.wait_for_page()
def _verify_and_return_staff_page(self):