diff --git a/data/instalocker.py b/data/instalocker.py index c6d2ada..a23b889 100644 --- a/data/instalocker.py +++ b/data/instalocker.py @@ -2945,11 +2945,11 @@ def find_game_end(self): self.pixel_patterns["main_menu"], ) - # red_button = self.compare_screenshot_to_pattern( - # self.locking_screenshotter, - # self.coords["play_button"], - # self.pixel_patterns["red_button"], - # ) + red_button = self.compare_screenshot_to_pattern( + self.locking_screenshotter, + self.coords["play_button"], + self.pixel_patterns["red_button"], + ) progress_text = self.compare_screenshot_to_pattern( self.locking_screenshotter, @@ -2963,7 +2963,7 @@ def find_game_end(self): self.pixel_patterns["pure_white"], ) - if player_banner or progress_text or progress_text_ranked: # or red_button: + if player_banner or red_button or progress_text or progress_text_ranked: confirmations += 1 if confirmations >= self.menu_screen_confirmaions_required: