From a98b027f0fb27c4720f28febfff469c4165227e0 Mon Sep 17 00:00:00 2001
From: Sollace <sollacea@gmail.com>
Date: Sun, 17 Feb 2019 19:21:22 +0200
Subject: [PATCH] Fixed spell entities not saving their affinity

---
 .../java/com/minelittlepony/unicopia/entity/EntitySpell.java | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/main/java/com/minelittlepony/unicopia/entity/EntitySpell.java b/src/main/java/com/minelittlepony/unicopia/entity/EntitySpell.java
index 11c61af3..6d510e6e 100644
--- a/src/main/java/com/minelittlepony/unicopia/entity/EntitySpell.java
+++ b/src/main/java/com/minelittlepony/unicopia/entity/EntitySpell.java
@@ -344,6 +344,10 @@ public class EntitySpell extends EntityCreature implements IMagicals, ICaster<En
 	@Override
 	public void readEntityFromNBT(NBTTagCompound compound) {
 		super.readEntityFromNBT(compound);
+		if (compound.hasKey("affinity")) {
+		    setAffinity(SpellAffinity.of(compound.getString("affinity")));
+		}
+
 		setOwner(compound.getString("ownerName"));
         setCurrentLevel(compound.getInteger("level"));
 
@@ -356,6 +360,7 @@ public class EntitySpell extends EntityCreature implements IMagicals, ICaster<En
 	public void writeEntityToNBT(NBTTagCompound compound) {
 		super.writeEntityToNBT(compound);
 
+		compound.setString("affinity", getAffinity().name());
         compound.setString("ownerName", getOwnerName());
         compound.setInteger("level", getCurrentLevel());