From 75a63e7a97163c4d042b853378217b4333769df2 Mon Sep 17 00:00:00 2001 From: Sollace Date: Sat, 26 Mar 2022 21:51:28 +0200 Subject: [PATCH] Fixed crash due to skeletons stealing items/having their bows stolen. Closes #53 --- .../unicopia/entity/ai/WantItTakeItGoal.java | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/src/main/java/com/minelittlepony/unicopia/entity/ai/WantItTakeItGoal.java b/src/main/java/com/minelittlepony/unicopia/entity/ai/WantItTakeItGoal.java index 7bed875e..a458ae21 100644 --- a/src/main/java/com/minelittlepony/unicopia/entity/ai/WantItTakeItGoal.java +++ b/src/main/java/com/minelittlepony/unicopia/entity/ai/WantItTakeItGoal.java @@ -62,12 +62,14 @@ public class WantItTakeItGoal extends BreakHeartGoal { mob.swingHand(Hand.MAIN_HAND); if (mob.world.random.nextInt(20) == 0) { - for (EquipmentSlot slot : EquipmentSlot.values()) { ItemStack stack = ((LivingEntity)target).getEquippedStack(slot); if (EnchantmentHelper.getLevel(UEnchantments.WANT_IT_NEED_IT, stack) > 0) { - target.equipStack(slot, ItemStack.EMPTY); - AwaitTickQueue.scheduleTask(mob.world, w -> mob.tryEquip(stack)); + AwaitTickQueue.scheduleTask(mob.world, w -> { + target.equipStack(slot, ItemStack.EMPTY); + mob.tryEquip(stack); + }); + break; } } }