From 02e0882900b90d684e4336b0112321e9105a6dbe Mon Sep 17 00:00:00 2001
From: RedDeadDepresso <94017243+RedDeadDepresso@users.noreply.github.com>
Date: Sat, 6 Jan 2024 22:37:18 +0000
Subject: [PATCH] fix: schedule popups

---
 tasks/schedule/ui.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tasks/schedule/ui.py b/tasks/schedule/ui.py
index d895d34..dfc0e09 100644
--- a/tasks/schedule/ui.py
+++ b/tasks/schedule/ui.py
@@ -18,7 +18,8 @@ class ScheduleUI(UI):
         while 1:
             self.device.screenshot()
             self.appear_then_click(dest_enter, interval=1)
-            self.handle_affection_level_up()
+            if self.handle_affection_level_up() or self.handle_location_level_up():
+                timer.reset()
             if self.appear(dest_check):
                 return True