fix visual glitch with removing stacked generals after retreat

1 files changed, 2 insertions, 0 deletions

diff --git a/rules.js b/rules.js
index 05caf09..0c3d23c 100644
--- a/rules.js
+++ b/rules.js
@@ -1066,6 +1066,8 @@ states.remove_general_after_retreat = {
 	general(g) {
 		push_undo()
 		action_remove_general(g)
+		if (g === game.move.who)
+			game.move.who = NOBODY
 		delete game.where
 		game.state = "retreat_after_battle_confirm"
 	},