Fixed disguiseNextEntity

src/main/java/me/libraryaddict/disguise/utilities/packets/packethandlers/PacketHandlerSpawn.java

@@ -70,10 +70,6 @@ public class PacketHandlerSpawn implements IPacketHandler {
     private void constructSpawnPackets(final Player observer, LibsPackets packets, Entity disguisedEntity) {
         Disguise disguise = packets.getDisguise();
 
-        if (disguise.getEntity() == null) {
-            disguise.setEntity(disguisedEntity);
-        }
-
         if (DisguiseConfig.isMiscDisguisesForLivingEnabled()) {
             if (disguise.getWatcher() instanceof LivingWatcher) {
 

src/main/java/me/libraryaddict/disguise/utilities/packets/packetlisteners/PacketListenerMain.java

@@ -12,6 +12,7 @@ import me.libraryaddict.disguise.disguisetypes.Disguise;
 import me.libraryaddict.disguise.utilities.DisguiseUtilities;
 import me.libraryaddict.disguise.utilities.packets.LibsPackets;
 import me.libraryaddict.disguise.utilities.packets.PacketsManager;
+import org.bukkit.entity.Entity;
 import org.bukkit.entity.Player;
 
 import java.lang.reflect.InvocationTargetException;
@@ -38,10 +39,25 @@ public class PacketListenerMain extends PacketAdapter {
 
         final Disguise disguise = DisguiseUtilities.getDisguise(observer, entityId);
 
+        if (disguise == null) {
+            return;
+        }
+
+        if (disguise.getEntity() == null) {
+            for (Entity e : observer.getWorld().getEntities()) {
+                if (e.getEntityId() != entityId) {
+                    continue;
+                }
+
+                disguise.setEntity(e);
+                break;
+            }
+        }
+
         // If the entity is the same as the sender. Don't disguise!
         // Prevents problems and there is no advantage to be gained.
         // Or if they are null and there's no disguise
-        if (disguise == null || disguise.getEntity() == observer) {
+        if (disguise.getEntity() == observer) {
             return;
         }
 

src/main/java/me/libraryaddict/disguise/utilities/packets/packetlisteners/PacketListenerSounds.java

@@ -73,7 +73,7 @@ public class PacketListenerSounds extends PacketAdapter {
                 for (TargetedDisguise entityDisguise : disguises) {
                     Entity entity = entityDisguise.getEntity();
 
-                    if (entity.getWorld() != observer.getWorld()) {
+                    if (entity == null || entity.getWorld() != observer.getWorld()) {
                         continue;
                     }