diff --git a/tests/test_feishu_media_message.py b/tests/test_feishu_media_message.py
index 02c5d73e6..9226633ff 100644
--- a/tests/test_feishu_media_message.py
+++ b/tests/test_feishu_media_message.py
@@ -8,7 +8,7 @@ ensure_optional_stub("Pinyin2Hanzi", is_pinyin=lambda value: False)
 
 from app.domain.context import MediaInfo  # noqa: E402
 from app.modules.feishu.feishu import Feishu  # noqa: E402
-from app.schemas import Message  # noqa: E402
+from app.schemas.message import Message  # noqa: E402
 
 
 def _build_feishu_client() -> Feishu: