Correctly switch classes in edit mode button when resetting edit mode