diff --git a/cms/static/coffee/src/views/module_edit.coffee b/cms/static/coffee/src/views/module_edit.coffee
index 83298a215b..a6f93197f3 100644
--- a/cms/static/coffee/src/views/module_edit.coffee
+++ b/cms/static/coffee/src/views/module_edit.coffee
@@ -86,5 +86,5 @@ class CMS.Views.ModuleEdit extends Backbone.View
 
   clickModeButton: (event) ->
     event.preventDefault()
-    @$el.removeClass('not-set')
-    @$el.addClass('is-set')
+    @$el.find("a").removeClass('not-set')
+    @$el.find("a").addClass('is-set')