diff --git a/src/main/java/net/torocraft/torohealth/util/HoldingWeaponUpdater.java b/src/main/java/net/torocraft/torohealth/util/HoldingWeaponUpdater.java index 1f91cc50..810de0eb 100644 --- a/src/main/java/net/torocraft/torohealth/util/HoldingWeaponUpdater.java +++ b/src/main/java/net/torocraft/torohealth/util/HoldingWeaponUpdater.java @@ -2,10 +2,15 @@ import net.minecraft.client.MinecraftClient; import net.minecraft.entity.player.PlayerEntity; -import net.minecraft.item.BowItem; +import net.minecraft.item.Item; import net.minecraft.item.ItemStack; -import net.minecraft.item.PotionItem; +import net.minecraft.item.AxeItem; +import net.minecraft.item.BowItem; +import net.minecraft.item.CrossbowItem; import net.minecraft.item.SwordItem; +import net.minecraft.item.ThrowablePotionItem; +import net.minecraft.item.TridentItem; + import net.torocraft.torohealth.ToroHealth; import net.torocraft.torohealth.config.Config.Mode; @@ -23,8 +28,13 @@ public static void update() { isWeapon(player.getMainHandStack()) || isWeapon(player.getOffHandStack()); } - private static boolean isWeapon(ItemStack item) { - return item.getItem() instanceof SwordItem || item.getItem() instanceof BowItem - || item.getItem() instanceof PotionItem; + private static boolean isWeapon(ItemStack itemStack) { + Item item = itemStack.getItem(); + return item instanceof AxeItem + || item instanceof BowItem + || item instanceof CrossbowItem + || item instanceof SwordItem + || item instanceof ThrowablePotionItem + || item instanceof TridentItem; } }