diff --git a/cms/djangoapps/contentstore/features/component_settings_editor_helpers.py b/cms/djangoapps/contentstore/features/component_settings_editor_helpers.py
index 6f572e9ae6..0a342f08e8 100644
--- a/cms/djangoapps/contentstore/features/component_settings_editor_helpers.py
+++ b/cms/djangoapps/contentstore/features/component_settings_editor_helpers.py
@@ -129,8 +129,8 @@ def edit_component(index=0):
# Verify that the "loading" indication has been hidden.
world.wait_for_loading()
# Verify that the "edit" button is present.
- world.wait_for(lambda _driver: world.css_visible('a.edit-button'))
- world.css_click('a.edit-button', index)
+ world.wait_for(lambda _driver: world.css_visible('.edit-button'))
+ world.css_click('.edit-button', index)
world.wait_for_ajax_complete()
diff --git a/cms/djangoapps/contentstore/features/pages.py b/cms/djangoapps/contentstore/features/pages.py
index bb3e113b64..f16aec9d60 100644
--- a/cms/djangoapps/contentstore/features/pages.py
+++ b/cms/djangoapps/contentstore/features/pages.py
@@ -38,7 +38,7 @@ def not_see_any_static_pages(step):
@step(u'I "(edit|delete)" the static page$')
def click_edit_or_delete(step, edit_or_delete):
- button_css = 'ul.component-actions a.%s-button' % edit_or_delete
+ button_css = 'ul.component-actions .%s-button' % edit_or_delete
world.css_click(button_css)
diff --git a/cms/djangoapps/contentstore/views/tests/utils.py b/cms/djangoapps/contentstore/views/tests/utils.py
index 094a789214..00358d005f 100644
--- a/cms/djangoapps/contentstore/views/tests/utils.py
+++ b/cms/djangoapps/contentstore/views/tests/utils.py
@@ -54,9 +54,10 @@ class StudioPageTestCase(CourseTestCase):
# Verify that there are no action buttons for public blocks
expected_button_html = [
- '',
- '',
- ''
+ '