1 files changed, 11 insertions, 1 deletions
diff --git a/bench/bench_window.cpp b/bench/bench_window.cpp
index 08f2236..39992eb 100644
--- a/bench/bench_window.cpp
+++ b/bench/bench_window.cpp
@@ -320,7 +320,17 @@ void bench_window::on_click_remove_warrior() {
         break;
       }
 
-    on_click_new_round();
+    if (warriors.size())
+      on_click_new_round();
+
+    else {
+      lib94::remove_all_warriors();
+      lib94::clear_address_sets();
+      core.mut.lock();
+      core.clear_all();
+      core.mut.unlock();
+      update_ui();
+    }
 
     delete w;