2 files changed, 19 insertions, 1 deletions
diff --git a/src/main/java/kubatech/api/network/CustomTileEntityPacket.java b/src/main/java/kubatech/api/network/CustomTileEntityPacket.java
index 9236594647..cd7bfd3389 100644
--- a/src/main/java/kubatech/api/network/CustomTileEntityPacket.java
+++ b/src/main/java/kubatech/api/network/CustomTileEntityPacket.java
@@ -59,6 +59,10 @@ public class CustomTileEntityPacket implements IMessage {
         addData(bytes);
     }
 
+    public void addData(boolean data) {
+        customdata.writeBoolean(data);
+    }
+
     public void getData(byte[] bytes) {
         customdata.readBytes(bytes);
     }
@@ -77,6 +81,10 @@ public class CustomTileEntityPacket implements IMessage {
         return new String(getData(getDataInt()), StandardCharsets.UTF_8);
     }
 
+    public boolean getDataBoolean() {
+        return customdata.readBoolean();
+    }
+
     @Override
     public void fromBytes(ByteBuf buf) {
         w = buf.readInt();
diff --git a/src/main/java/kubatech/api/utils/MobUtils.java b/src/main/java/kubatech/api/utils/MobUtils.java
index 3e20adac1c..0b20364484 100644
--- a/src/main/java/kubatech/api/utils/MobUtils.java
+++ b/src/main/java/kubatech/api/utils/MobUtils.java
@@ -17,6 +17,16 @@ public class MobUtils {
 
     @SideOnly(Side.CLIENT)
     public static float getDesiredScale(EntityLiving e, float desiredHeight) {
+        return getDesiredScale(getMobHeight(e), desiredHeight);
+    }
+
+    @SideOnly(Side.CLIENT)
+    public static float getDesiredScale(float entityHeight, float desiredHeight) {
+        return desiredHeight / entityHeight;
+    }
+
+    @SideOnly(Side.CLIENT)
+    public static float getMobHeight(EntityLiving e) {
         try {
             if (mainmodelfield == null) {
                 mainmodelfield = RendererLivingEntity.class.getDeclaredField(
@@ -49,7 +59,7 @@ public class MobUtils {
                     }
                 }
             }
-            return desiredHeight / eheight;
+            return eheight;
         } catch (Exception ex) {
             return 1f;
         }