Added missing EmailMessage import