diff --git a/src/main/java/com/minelittlepony/unicopia/client/minelittlepony/Main.java b/src/main/java/com/minelittlepony/unicopia/client/minelittlepony/Main.java index 7a02815f..ee14a709 100644 --- a/src/main/java/com/minelittlepony/unicopia/client/minelittlepony/Main.java +++ b/src/main/java/com/minelittlepony/unicopia/client/minelittlepony/Main.java @@ -37,13 +37,12 @@ public class Main extends MineLPDelegate implements ClientModInitializer { IGear.register(GlassesGear::new); } - @SuppressWarnings("unchecked") private void onPonyModelPrepared(Entity entity, IModel model, ModelAttributes.Mode mode) { if (hookErroring) return; try { if (entity instanceof PlayerEntity) { - if (entity instanceof Owned) { - entity = ((Owned)entity).getMaster(); + if (entity instanceof Owned o && o.getMaster() instanceof PlayerEntity master) { + entity = master; } Pony pony = Pony.of((PlayerEntity)entity);