fix supply train breakage in previous commit

1 files changed, 9 insertions, 4 deletions

diff --git a/rules.js b/rules.js
index 701044e..f0b082d 100644
--- a/rules.js
+++ b/rules.js
@@ -1352,11 +1352,16 @@ states.movement = {
 	piece(p) {
 		push_undo()
 
-		game.selected = []
 		let here = game.pos[p]
-		for (let other of all_power_generals[game.power])
-			if (other >= p && game.pos[other] === here && !set_has(game.moved, other))
-				game.selected.push(other)
+
+		if (is_general(p)) {
+			game.selected = []
+			for (let other of all_power_generals[game.power])
+				if (other >= p && game.pos[other] === here && !set_has(game.moved, other))
+					game.selected.push(other)
+		} else {
+			game.selected = [ p ]
+		}
 
 		game.count = 0