minor adjustment to when we remove the editing class from messages.