From ac04e2aace65d5e5ca909a68f13e6ad05c2fdde5 Mon Sep 17 00:00:00 2001
From: LmeSzinc <37934724+LmeSzinc@users.noreply.github.com>
Date: Wed, 19 Jul 2023 13:36:48 +0800
Subject: [PATCH] Fix: Check MAIN_GOTO_BATTLE_PASS

---
 tasks/battle_pass/battle_pass.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tasks/battle_pass/battle_pass.py b/tasks/battle_pass/battle_pass.py
index 927cdcd90..cc4658d34 100644
--- a/tasks/battle_pass/battle_pass.py
+++ b/tasks/battle_pass/battle_pass.py
@@ -8,7 +8,7 @@ from module.config.utils import get_server_next_update
 from module.logger.logger import logger
 from module.ocr.ocr import Digit, Duration
 from module.ui.switch import Switch
-from tasks.base.assets.assets_base_page import BATTLE_PASS_CHECK
+from tasks.base.assets.assets_base_page import BATTLE_PASS_CHECK, MAIN_GOTO_BATTLE_PASS
 from tasks.base.assets.assets_base_popup import GET_REWARD
 from tasks.base.page import page_battle_pass, page_main
 from tasks.base.ui import UI
@@ -183,7 +183,7 @@ class BattlePassUI(UI):
 
     def run(self):
         self.ui_ensure(page_main)
-        if not self.appear(page_battle_pass.check_button):
+        if not self.appear(MAIN_GOTO_BATTLE_PASS):
             logger.info('No battle pass entrance, probably a gap between two periods')
             self.config.task_delay(server_update=True)
             self.config.task_stop()