From 9f185cd8a8def1a838ed6c3842e3f210675bf206 Mon Sep 17 00:00:00 2001
From: Bertie690 <taylormw163@gmail.com>
Date: Wed, 3 Sep 2025 21:03:11 -0400
Subject: [PATCH] fixed the bug, ruined my sanity

---
 test/test-utils/helpers/prompt-handler.ts | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/test/test-utils/helpers/prompt-handler.ts b/test/test-utils/helpers/prompt-handler.ts
index 4b16636f4db..c727fa87616 100644
--- a/test/test-utils/helpers/prompt-handler.ts
+++ b/test/test-utils/helpers/prompt-handler.ts
@@ -77,14 +77,18 @@ export class PromptHandler extends GameManagerHelper {
    * Helper method to wrap UI mode changing.
    * @param args - Arguments being passed to the original method
    * @returns The original return value.
+   * @todo Make this wait for the actual UI mode setting
    */
-  private async setMode(args: Parameters<typeof this.originalSetModeInternal>) {
+  private setMode(
+    args: Parameters<typeof this.originalSetModeInternal>,
+  ): ReturnType<typeof this.originalSetModeInternal> {
     const mode = args[0];
 
     this.doLog(
       `UI mode changed from ${getEnumStr(UiMode, this.game.scene.ui.getMode())} to ${getEnumStr(UiMode, mode)}!`,
     );
-    const ret = await this.originalSetModeInternal.apply(this.game.scene.ui, args);
+    // TODO: Add `await` to this
+    const ret = this.originalSetModeInternal.apply(this.game.scene.ui, args);
 
     const currentPhase = this.game.scene.phaseManager.getCurrentPhase()?.phaseName!;
     if (endBySetMode.includes(currentPhase)) {