Fix shaded 5 unit selection

2 files changed, 2 insertions, 2 deletions

diff --git a/events.txt b/events.txt
index 8d6f306..fa1387d 100644
--- a/events.txt
+++ b/events.txt
@@ -112,7 +112,7 @@ SHADED 5
 	stay_eligible
 	current DS
 	prompt "Replace up to 3 opposing Units adjacent to Chittor."
-	piece_undo_opt 3 (is_adjacent_to_city(C_CHITTOR, piece_space(p)) && is_enemy_piece(p))
+	piece_undo_opt 3 (is_adjacent_to_city(C_CHITTOR, piece_space(p)) && is_enemy_unit(p))
 		replace DS TROOPS
 	endpiece
 	prompt "Place a Qasbah in Rajput Kingdoms."
diff --git a/rules.js b/rules.js
index aeacbe5..63498aa 100644
--- a/rules.js
+++ b/rules.js
@@ -6026,7 +6026,7 @@ CODE[5 * 2 + 1] = [
 	[ vm_stay_eligible ],
 	[ vm_current, DS ],
 	[ vm_prompt, "Replace up to 3 opposing Units adjacent to Chittor." ],
-	[ vm_piece, true, 0, 3, (p,s)=>(is_adjacent_to_city(C_CHITTOR, piece_space(p)) && is_enemy_piece(p)) ],
+	[ vm_piece, true, 0, 3, (p,s)=>(is_adjacent_to_city(C_CHITTOR, piece_space(p)) && is_enemy_unit(p)) ],
 	[ vm_replace, DS, TROOPS, false, false ],
 	[ vm_endpiece ],
 	[ vm_prompt, "Place a Qasbah in Rajput Kingdoms." ],