From ab442461d34bdb5ffdfbea4f5d39a77f988a0157 Mon Sep 17 00:00:00 2001
From: Sollace <sollacea@gmail.com>
Date: Wed, 18 Aug 2021 20:13:52 +0200
Subject: [PATCH] Fixed disguises not being rotated properly when using mine
 little pony

---
 .../minelittlepony/unicopia/client/minelittlepony/Main.java  | 5 +++++
 1 file changed, 5 insertions(+)

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 23a1707e..6a099775 100644
--- a/src/main/java/com/minelittlepony/unicopia/client/minelittlepony/Main.java
+++ b/src/main/java/com/minelittlepony/unicopia/client/minelittlepony/Main.java
@@ -3,6 +3,7 @@ package com.minelittlepony.unicopia.client.minelittlepony;
 import com.minelittlepony.api.model.IModel;
 import com.minelittlepony.api.model.ModelAttributes;
 import com.minelittlepony.api.model.fabric.PonyModelPrepareCallback;
+import com.minelittlepony.unicopia.Owned;
 import com.minelittlepony.unicopia.Unicopia;
 import com.minelittlepony.unicopia.entity.player.Pony;
 
@@ -20,10 +21,14 @@ public class Main implements ClientModInitializer {
         PonyModelPrepareCallback.EVENT.register(this::onPonyModelPrepared);
     }
 
+    @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<PlayerEntity>)entity).getMaster();
+                }
                 Pony pony = Pony.of((PlayerEntity)entity);
 
                 if (pony.getMotion().isFlying()) {