diff --git a/Mage/src/main/java/mage/game/permanent/token/ZombieArmyToken.java b/Mage/src/main/java/mage/game/permanent/token/ZombieArmyToken.java index 86fe1a490d1..dda369ef8ce 100644 --- a/Mage/src/main/java/mage/game/permanent/token/ZombieArmyToken.java +++ b/Mage/src/main/java/mage/game/permanent/token/ZombieArmyToken.java @@ -1,9 +1,11 @@ - package mage.game.permanent.token; import mage.MageInt; import mage.constants.CardType; import mage.constants.SubType; +import mage.util.RandomUtil; + +import java.util.Arrays; /** * @author TheElk801 @@ -12,13 +14,15 @@ public final class ZombieArmyToken extends TokenImpl { public ZombieArmyToken() { super("Zombie Army", "0/0 black Zombie Army creature token"); - setExpansionSetCodeForImage("WAR"); // default + cardType.add(CardType.CREATURE); color.setBlack(true); subtype.add(SubType.ZOMBIE); subtype.add(SubType.ARMY); power = new MageInt(0); toughness = new MageInt(0); + + availableImageSetCodes = Arrays.asList("WAR", "MH2"); } private ZombieArmyToken(final ZombieArmyToken token) { @@ -29,4 +33,13 @@ public final class ZombieArmyToken extends TokenImpl { public ZombieArmyToken copy() { return new ZombieArmyToken(this); } + + @Override + public void setExpansionSetCodeForImage(String code) { + super.setExpansionSetCodeForImage(code); + + if (getOriginalExpansionSetCode() != null && getOriginalExpansionSetCode().equals("WAR")) { + this.setTokenType(RandomUtil.nextInt(3) + 1); // 1..3 + } + } }