diff --git a/tests/test_plugins/test_feishu.py b/tests/test_plugins/test_feishu.py
index 54a0d7ae..960ae377 100644
--- a/tests/test_plugins/test_feishu.py
+++ b/tests/test_plugins/test_feishu.py
@@ -1,12 +1,11 @@
-from nonebot.adapters.feishu.event import MessageEvent
-from nonebot.rule import to_me
 from nonebot.plugin import on_command
-from nonebot.adapters.feishu import Bot as FeishuBot, MessageSegment, MessageEvent
+from nonebot.typing import T_State
+from nonebot.adapters.feishu import Bot as FeishuBot, MessageEvent
 
 helper = on_command("say")
 
 
 @helper.handle()
-async def feishu_helper(bot: FeishuBot, event: MessageEvent):
+async def feishu_helper(bot: FeishuBot, event: MessageEvent, state: T_State):
     message = event.get_message()
     await helper.finish(message, at_sender=True)