diff --git a/src/main/java/com/minelittlepony/unicopia/ability/magic/spell/effect/CustomisedSpellType.java b/src/main/java/com/minelittlepony/unicopia/ability/magic/spell/effect/CustomisedSpellType.java index 00a25c09..2399dfcb 100644 --- a/src/main/java/com/minelittlepony/unicopia/ability/magic/spell/effect/CustomisedSpellType.java +++ b/src/main/java/com/minelittlepony/unicopia/ability/magic/spell/effect/CustomisedSpellType.java @@ -31,11 +31,11 @@ public record CustomisedSpellType ( @Override public boolean test(Spell spell) { - return type.test(spell); + return type.test(spell) && spell.getTraits().equals(traits); } public ItemStack getDefaultStack() { - return type.getDefualtStack(); + return traits.applyTo(type.getDefualtStack()); } public TypedActionResult> toAction() {