From ca774de781f5573950ad900990dcb14a4dc1015f Mon Sep 17 00:00:00 2001
From: Mischa Untaga <99098079+MischaU8@users.noreply.github.com>
Date: Thu, 26 Oct 2023 20:54:11 +0200
Subject: undo whole airmobilize submenu as a whole

---
 rules.js | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/rules.js b/rules.js
index 79258a7..38de71f 100644
--- a/rules.js
+++ b/rules.js
@@ -4020,6 +4020,8 @@ states.gov_flush_select_units = {
 
 function goto_gov_airmobilize() {
 	game.selected = []
+	push_undo()
+
 	game.from_state = game.state
 	game.state = "gov_airmobilize_select_units"
 }
@@ -4051,7 +4053,6 @@ states.gov_airmobilize_select_units = {
 		gen_action("done")
 	},
 	unit(u) {
-		push_undo()
 		set_toggle(game.selected, u)
 		// preview selection to see backside of counter
 		if (set_has(game.selected, u)) {
@@ -4064,7 +4065,6 @@ states.gov_airmobilize_select_units = {
 		let list = game.selected
 		game.selected = []
 
-		push_undo()
 		let cost = airmobilize_cost(list)
 		game.helo_avail -= cost
 		log(`Airmobilized (using ${cost} Helo PTS):`)
-- 
cgit v1.2.3