diff --git a/src/main/java/electroblob/wizardry/entity/living/EntityEvilWizard.java b/src/main/java/electroblob/wizardry/entity/living/EntityEvilWizard.java index 30cb09d8..020db8fb 100644 --- a/src/main/java/electroblob/wizardry/entity/living/EntityEvilWizard.java +++ b/src/main/java/electroblob/wizardry/entity/living/EntityEvilWizard.java @@ -118,10 +118,10 @@ public class EntityEvilWizard extends EntityMob implements ISpellCaster, IEntity || (entity instanceof ISummonedCreature || entity instanceof EntityWizard // ... or in the whitelist ... || Arrays.asList(Wizardry.settings.summonedCreatureTargetsWhitelist) - .contains(EntityList.getEntityString(entity).toLowerCase(Locale.ROOT))) + .contains(EntityList.getKey(entity.getClass()).toString().toLowerCase(Locale.ROOT))) // ... and isn't in the blacklist ... && !Arrays.asList(Wizardry.settings.summonedCreatureTargetsBlacklist) - .contains(EntityList.getEntityString(entity).toLowerCase(Locale.ROOT))){ + .contains(EntityList.getKey(entity.getClass()).toString().toLowerCase(Locale.ROOT))){ // ... it can be attacked. return true; } diff --git a/src/main/java/electroblob/wizardry/entity/living/EntityWizard.java b/src/main/java/electroblob/wizardry/entity/living/EntityWizard.java index b34c5a5f..eb7ad7d3 100644 --- a/src/main/java/electroblob/wizardry/entity/living/EntityWizard.java +++ b/src/main/java/electroblob/wizardry/entity/living/EntityWizard.java @@ -162,10 +162,10 @@ public class EntityWizard extends EntityVillager implements ISpellCaster, IEntit if((entity instanceof IMob || entity instanceof ISummonedCreature // ... or in the whitelist ... || Arrays.asList(Wizardry.settings.summonedCreatureTargetsWhitelist) - .contains(EntityList.getEntityString(entity).toLowerCase(Locale.ROOT))) + .contains(EntityList.getKey(entity.getClass()).toString().toLowerCase(Locale.ROOT))) // ... and isn't in the blacklist ... && !Arrays.asList(Wizardry.settings.summonedCreatureTargetsBlacklist) - .contains(EntityList.getEntityString(entity).toLowerCase(Locale.ROOT))){ + .contains(EntityList.getKey(entity.getClass()).toString().toLowerCase(Locale.ROOT))){ // ... it can be attacked. return true; } diff --git a/src/main/java/electroblob/wizardry/entity/living/ISummonedCreature.java b/src/main/java/electroblob/wizardry/entity/living/ISummonedCreature.java index 792d6382..092a7dd5 100644 --- a/src/main/java/electroblob/wizardry/entity/living/ISummonedCreature.java +++ b/src/main/java/electroblob/wizardry/entity/living/ISummonedCreature.java @@ -190,10 +190,10 @@ public interface ISummonedCreature extends IEntityAdditionalSpawnData { || (entity instanceof EntityWizard && !(getCaster() instanceof EntityWizard)) // ... or in the whitelist ... || Arrays.asList(Wizardry.settings.summonedCreatureTargetsWhitelist) - .contains(EntityList.getEntityString(entity).toLowerCase(Locale.ROOT))) + .contains(EntityList.getKey(entity.getClass()).toString().toLowerCase(Locale.ROOT))) // ... and isn't in the blacklist ... && !Arrays.asList(Wizardry.settings.summonedCreatureTargetsBlacklist) - .contains(EntityList.getEntityString(entity).toLowerCase(Locale.ROOT))){ + .contains(EntityList.getKey(entity.getClass()).toString().toLowerCase(Locale.ROOT))){ // ... it can be attacked. return true; }