From 0fde96b43bb2426e45e3730774ff153157f5c32a Mon Sep 17 00:00:00 2001
From: Sense101 <67970865+Sense101@users.noreply.github.com>
Date: Tue, 11 May 2021 15:48:37 +0100
Subject: [PATCH] improved menu return

---
 src/js/game/hud/parts/puzzle_complete_notification.js | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/js/game/hud/parts/puzzle_complete_notification.js b/src/js/game/hud/parts/puzzle_complete_notification.js
index 9e093c19..34d600bb 100644
--- a/src/js/game/hud/parts/puzzle_complete_notification.js
+++ b/src/js/game/hud/parts/puzzle_complete_notification.js
@@ -93,9 +93,10 @@ export class HUDPuzzleCompleteNotification extends BaseHUDPart {
             .then(() => {
                 if (toMenu) {
                     this.root.gameState.moveToState("PuzzleMenuState");
+                } else {
+                    this.visible = false;
+                    this.cleanup();
                 }
-                this.visible = false;
-                this.cleanup();
             });
     }