Fixed problem with inproper classes when mail is in edit mode