From 5adbde4f33b419e76e65fb4e682702a13423fb1d Mon Sep 17 00:00:00 2001
From: LmeSzinc <37934724+LmeSzinc@users.noreply.github.com>
Date: Sun, 4 Aug 2024 09:54:34 +0800
Subject: [PATCH] Fix: combat_set_wave has no screenshot updated

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

diff --git a/tasks/combat/prepare.py b/tasks/combat/prepare.py
index 98dcf718a..3f76ecd87 100644
--- a/tasks/combat/prepare.py
+++ b/tasks/combat/prepare.py
@@ -131,7 +131,8 @@ class CombatPrepare(UI):
                     logger.warning(f'Combat wave costs {cost} but has multiple waves, '
                                    f'probably wave amount is preset')
                     self.combat_set_wave(1)
-                    skip_first_screenshot = True
+                    # Don't skip_first_screenshot, combat_set_wave may not have screenshot updated
+                    # skip_first_screenshot = True
                     timeout.reset()
                     continue
                 else: