diff --git a/src/Action/TrainTrooperAction.ts b/src/Action/TrainTrooperAction.ts
index 8a9ca81..ef430ec 100644
--- a/src/Action/TrainTrooperAction.ts
+++ b/src/Action/TrainTrooperAction.ts
@@ -10,7 +10,7 @@ export class TrainTrooperAction extends ActionController {
         const troopId = this.getTroopId(args);
         const trainCount = this.getTrainCount(args);
 
-        const block = jQuery(`#nonFavouriteTroops .innerTroopWrapper.troop${troopId}`);
+        const block = jQuery(`.innerTroopWrapper[data-troopid="${troopId}"]`);
         if (block.length !== 1) {
             throw new ActionError(`Troop block not found`);
         }