diff --git a/src/main/java/electroblob/wizardry/entity/living/EntityEvilWizard.java b/src/main/java/electroblob/wizardry/entity/living/EntityEvilWizard.java index 769f2baa..5648a941 100644 --- a/src/main/java/electroblob/wizardry/entity/living/EntityEvilWizard.java +++ b/src/main/java/electroblob/wizardry/entity/living/EntityEvilWizard.java @@ -278,7 +278,8 @@ public class EntityEvilWizard extends EntityMob implements ISpellCaster, IEntity @Override public void writeEntityToNBT(NBTTagCompound nbt){ super.writeEntityToNBT(nbt); - nbt.setInteger("element", this.getElement().ordinal()); + Element element = this.getElement(); + nbt.setInteger("element", element == null ? 0 : element.ordinal()); nbt.setInteger("skin", this.textureIndex); NBTExtras.storeTagSafely(nbt, "spells", NBTExtras.listToNBT(spells, spell -> new NBTTagInt(spell.metadata()))); nbt.setBoolean("hasStructure", this.hasStructure); diff --git a/src/main/java/electroblob/wizardry/entity/living/EntityWizard.java b/src/main/java/electroblob/wizardry/entity/living/EntityWizard.java index fce4bd33..69893d82 100644 --- a/src/main/java/electroblob/wizardry/entity/living/EntityWizard.java +++ b/src/main/java/electroblob/wizardry/entity/living/EntityWizard.java @@ -404,7 +404,8 @@ public class EntityWizard extends EntityCreature implements INpc, IMerchant, ISp NBTExtras.storeTagSafely(nbt, "trades", this.trades.getRecipiesAsTags()); } - nbt.setInteger("element", this.getElement().ordinal()); + Element element = this.getElement(); + nbt.setInteger("element", element == null ? 0 : element.ordinal()); nbt.setInteger("skin", this.textureIndex); NBTExtras.storeTagSafely(nbt, "spells", NBTExtras.listToNBT(spells, spell -> new NBTTagInt(spell.metadata())));