Fix retreat forced march and engaging state change back to retreat move.

1 files changed, 10 insertions, 3 deletions

diff --git a/rules.js b/rules.js
index 2ea7d19..f0717fb 100644
--- a/rules.js
+++ b/rules.js
@@ -3686,6 +3686,13 @@ function move_unit(who, to, speed, move) {
 	}
 }
 
+function resume_move() {
+	if (game.retreat > 0)
+		game.state = 'retreat_move'
+	else
+		game.state = 'move'
+}
+
 states.forced_march_via = {
 	inactive: "movement",
 	prompt() {
@@ -3713,7 +3720,7 @@ states.forced_march_via = {
 
 		forced_march_via(game.hexside.who, via, game.hexside.to, game.hexside.move)
 		game.hexside = null
-		game.state = 'move'
+		resume_move()
 	}
 }
 
@@ -3749,7 +3756,7 @@ states.engage_via = {
 
 		forced_march_via(game.hexside.who, via, game.hexside.to, game.hexside.move)
 		game.hexside = null
-		game.state = 'move'
+		resume_move()
 	},
 	hex(via) {
 		let rommel = (game.hexside.move === game.rommel) ? 1 : 0
@@ -3760,7 +3767,7 @@ states.engage_via = {
 
 		engage_via(game.hexside.who, via, game.hexside.to, game.hexside.move)
 		game.hexside = null
-		game.state = 'move'
+		resume_move()
 	}
 }