diff --git a/src/main/java/com/minelittlepony/unicopia/ability/Abilities.java b/src/main/java/com/minelittlepony/unicopia/ability/Abilities.java index 7bd91b83..97df71e0 100644 --- a/src/main/java/com/minelittlepony/unicopia/ability/Abilities.java +++ b/src/main/java/com/minelittlepony/unicopia/ability/Abilities.java @@ -15,8 +15,8 @@ public interface Abilities { Registry> REGISTRY = Registries.createSimple(new Identifier("unicopia", "abilities")); // unicorn / alicorn - Ability TELEPORT = register(new UnicornTeleportAbility(), "teleport", AbilitySlot.SECONDARY); Ability CAST = register(new UnicornCastingAbility(), "cast", AbilitySlot.PRIMARY); + Ability TELEPORT = register(new UnicornTeleportAbility(), "teleport", AbilitySlot.SECONDARY); // earth / alicorn Ability STOMP = register(new EarthPonyStompAbility(), "stomp", AbilitySlot.PRIMARY); diff --git a/src/main/java/com/minelittlepony/unicopia/ability/AbilityDispatcher.java b/src/main/java/com/minelittlepony/unicopia/ability/AbilityDispatcher.java index d6b8db40..f153910a 100644 --- a/src/main/java/com/minelittlepony/unicopia/ability/AbilityDispatcher.java +++ b/src/main/java/com/minelittlepony/unicopia/ability/AbilityDispatcher.java @@ -1,5 +1,6 @@ package com.minelittlepony.unicopia.ability; +import java.util.Collections; import java.util.EnumMap; import java.util.Map; import java.util.Optional; @@ -177,7 +178,7 @@ public class AbilityDispatcher implements Tickable, NbtSerialisable { public Optional> getAbility() { Race race = player.getSpecies(); - return Abilities.BY_SLOT.get(slot).stream().filter(a -> a.canUse(race)).findFirst(); + return Abilities.BY_SLOT.computeIfAbsent(slot, c -> Collections.emptySet()).stream().filter(a -> a.canUse(race)).findFirst(); } protected synchronized void setActiveAbility(Ability power) {