diff --git a/tests/unit/test_models.py b/tests/unit/test_models.py
index 36be62a..f8dc811 100644
--- a/tests/unit/test_models.py
+++ b/tests/unit/test_models.py
@@ -1,7 +1,7 @@
 from datetime import datetime, timedelta
 
-from mail_intake.models.message import MailIntakeMessage
-from mail_intake.models.thread import MailIntakeThread
+from mail_intake.models import MailIntakeMessage
+from mail_intake.models import MailIntakeThread
 
 
 def test_message_is_immutable():
diff --git a/tests/unit/test_parsers.py b/tests/unit/test_parsers.py
index 6c75491..026090f 100644
--- a/tests/unit/test_parsers.py
+++ b/tests/unit/test_parsers.py
@@ -1,8 +1,8 @@
 import base64
 
-from mail_intake.parsers.subject import normalize_subject
-from mail_intake.parsers.headers import parse_headers, extract_sender
-from mail_intake.parsers.body import extract_body
+from mail_intake.parsers import normalize_subject
+from mail_intake.parsers import parse_headers, extract_sender
+from mail_intake.parsers import extract_body
 
 
 def _b64(text: str) -> str: