1 files changed, 9 insertions, 4 deletions

diff --git a/play.js b/play.js
index f083057..1535599 100644
--- a/play.js
+++ b/play.js
@@ -173,7 +173,7 @@ function is_unit_moved(u) {
 }
 
 function is_unit_fired(u) {
-	return set_has(view.fired, u)
+	return set_has(view.fired, u) || set_has(view.retreat, u)
 }
 
 function is_unit_revealed(u) {
@@ -200,8 +200,12 @@ function is_unit_selected(unit) {
 	return view.selected === unit
 }
 
-function is_hex_action(hex) {
-	return !!(view.actions && view.actions.hex && view.actions.hex.includes(hex))
+function is_any_hex_action(hex) {
+	if (view.actions && view.actions.hex && view.actions.hex.includes(hex))
+		return true
+	if (is_hex_forced_march_action(hex))
+		return true
+	return false
 }
 
 function is_hex_forced_march_action(hex) {
@@ -847,7 +851,7 @@ function update_map() {
 		}
 
 		if (ui.hexes[hex]) {
-			ui.hexes[hex].classList.toggle("action", is_hex_action(hex) || is_hex_forced_march_action(hex))
+			ui.hexes[hex].classList.toggle("action", is_any_hex_action(hex))
 			ui.hexes[hex].classList.toggle("forced_march", is_hex_forced_march_action(hex))
 			ui.hexes[hex].classList.toggle("selected", is_hex_selected(hex))
 			ui.hexes[hex].classList.toggle("axis_control", is_hex_axis_controlled(hex))
@@ -993,6 +997,7 @@ function on_update() {
 
 	action_button("eliminate", "Eliminate")
 	action_button("overrun", "Overrun")
+	action_button("refuse", "Refuse")
 	action_button("retreat", "Retreat")
 	action_button("probe", "Probe")