diff --git a/main.py b/main.py
index 67ba1b3..26eb6e0 100644
--- a/main.py
+++ b/main.py
@@ -7,7 +7,10 @@ from .back import time_long, volume, isUserExist, insertUser, seconds_to_hms, ml
 import pymysql
 from .Tool import get_tool_name
 import astrbot.api.message_components as Comp
-from astrbot.session import session_waiter, SessionController
+from astrbot.core.utils.session_waiter import (
+    session_waiter,
+    SessionController,
+)
 import aiohttp  # 需要异步HTTP客户端
 from TicTacToeGame import TicTacToeGame