From dc95f5f0a2fa48379dd1dc3729fda4d4333c64d1 Mon Sep 17 00:00:00 2001
From: Aki <please@ignore.pl>
Date: Sat, 7 Jan 2023 16:48:22 +0100
Subject: Fixed one imgui frame showing after campaign ends

---
 kurator/src/Campaign.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/kurator/src/Campaign.cpp b/kurator/src/Campaign.cpp
index c0e28f1..be02862 100644
--- a/kurator/src/Campaign.cpp
+++ b/kurator/src/Campaign.cpp
@@ -52,6 +52,8 @@ Campaign::Campaign(std::shared_ptr<Session> _session) :
 void
 Campaign::update(const float)
 {
+	if (ships.empty())
+		return session->pop();
 	if (ImGui::Begin("Campaign")) {
 		ImGui::Text("Level %d, ships left:", level);
 		for (const auto& ship : ships) {
@@ -76,8 +78,6 @@ Campaign::update(const float)
 			session->pop();
 	}
 	ImGui::End();
-	if (ships.empty())
-		session->pop();
 }
 
 
-- 
cgit v1.1