diff --git a/tasks/shop/ui.py b/tasks/shop/ui.py index 58eefb7..54a548a 100644 --- a/tasks/shop/ui.py +++ b/tasks/shop/ui.py @@ -109,7 +109,7 @@ class ShopUI(UI): self.swipe_flags[8], self.swipe_flags[16] = False, False def make_purchase(self): - if self.select_then_check(PURCHASE, CONFIRM_PURCHASE) and self.appear_then_click(CONFIRM_PURCHASE): + if self.match_color(PURCHASE) and self.select_then_check(PURCHASE, CONFIRM_PURCHASE) and self.appear_then_click(CONFIRM_PURCHASE): return True logger.warning("No items were selected. Unable to purchase.") return False