diff --git a/tasks/shop/shop.py b/tasks/shop/shop.py index 3c639e3..50cd143 100644 --- a/tasks/shop/shop.py +++ b/tasks/shop/shop.py @@ -83,7 +83,7 @@ class Shop(ShopUI): return ShopStatus.SELECT_SHOP return ShopStatus.END case ShopStatus.END: - if self.appear(page_shop.check_button): + if self.appear(page_shop.check_button) and self.match_color(page_shop.check_button): self.task.pop(0) return ShopStatus.SELECT_SHOP self.click_with_interval(BACK, interval=2)