diff --git a/models/__init__.py b/models/__init__.py
index 7103dd1..a25a0af 100644
--- a/models/__init__.py
+++ b/models/__init__.py
@@ -1,5 +1,5 @@
 from .events.base import OneBotEvent
-from .events.message import MessageEvent, PrivateMessageEvent, GroupMessageEvent
+from .events.message import MessageEvent, PrivateMessageEvent, GroupMessageEvent, MessageSegment
 from .events.notice import (
     NoticeEvent, FriendAddNoticeEvent, FriendRecallNoticeEvent,
     GroupRecallNoticeEvent, GroupIncreaseNoticeEvent,