From ea5547a6c6e5d8a85b359575e15532db98eee6f3 Mon Sep 17 00:00:00 2001
From: Andrew <libraryaddict115@yahoo.co.nz>
Date: Wed, 7 Aug 2013 22:51:55 +1200
Subject: [PATCH] Add a sanity check

---
 src/me/libraryaddict/disguise/LibsDisguises.java | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/me/libraryaddict/disguise/LibsDisguises.java b/src/me/libraryaddict/disguise/LibsDisguises.java
index 19911be4..3478983c 100644
--- a/src/me/libraryaddict/disguise/LibsDisguises.java
+++ b/src/me/libraryaddict/disguise/LibsDisguises.java
@@ -132,6 +132,8 @@ public class LibsDisguises extends JavaPlugin {
     }
 
     protected PacketContainer[] constructPacket(Disguise disguise, Entity disguisedEntity) {
+        if (disguise.getEntity() == null)
+            disguise.setEntity(disguisedEntity);
         ProtocolManager manager = ProtocolLibrary.getProtocolManager();
         net.minecraft.server.v1_6_R2.Entity nmsEntity = ((CraftEntity) disguisedEntity).getHandle();
         ArrayList<PacketContainer> packets = new ArrayList<PacketContainer>();