diff --git a/TrayIcon.py b/TrayIcon.py
index afc577d..86b1f5d 100644
--- a/TrayIcon.py
+++ b/TrayIcon.py
@@ -49,7 +49,7 @@ def __init__(self):
             win32con.WM_DESTROY: self.OnDestroy,
             win32con.WM_COMMAND: self.OnCommand,
             win32con.WM_USER + 20: self.OnTaskbarNotify,
-            win32con.WM_ENDSESSION: self.OnEndSession
+            win32con.WM_QUERYENDSESSION: self.OnEndSession
         }
         # 注册窗口类
         wndclass = win32gui.WNDCLASS()
@@ -181,7 +181,7 @@ def OnEndSession(self, hwnd, msg, wparam, lparam):
         self.stopflag=True
         msg=self.legod.pause()
         self.taskbar_msg("退出并暂停时长结果",msg)
-        return 0
+        return True
     def taskbar_msg(self,title,msg):
         # Taskbar icon
         nid =self.nid[4]