diff --git a/src/email_providers/imap.py b/src/email_providers/imap.py
index 4d2994b..334a600 100644
--- a/src/email_providers/imap.py
+++ b/src/email_providers/imap.py
@@ -174,7 +174,7 @@ class IMAPProvider(BaseProvider):
             logger.debug(f"Error parsing message: {e}")
             return None
 
-    def _get_body(self, msg: email.message.Message) -> str:
+    def _get_body(self, msg: Message) -> str:
         """Extract email body."""
         body = ""
 
@@ -195,7 +195,7 @@ class IMAPProvider(BaseProvider):
 
         return body if isinstance(body, str) else str(body)
 
-    def _parse_attachments(self, msg: email.message.Message) -> List[Attachment]:
+    def _parse_attachments(self, msg: Message) -> List[Attachment]:
         """Extract attachment metadata."""
         attachments = []