From 46e4d14bb0a52b8a79635f4097760589011a823d Mon Sep 17 00:00:00 2001
From: Sollace <sollacea@gmail.com>
Date: Thu, 1 Sep 2022 22:57:19 +0200
Subject: [PATCH] Whoops!

---
 .../minelittlepony/unicopia/entity/CastSpellEntity.java   | 5 +++++
 .../java/com/minelittlepony/unicopia/entity/Creature.java | 8 ++++++--
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/src/main/java/com/minelittlepony/unicopia/entity/CastSpellEntity.java b/src/main/java/com/minelittlepony/unicopia/entity/CastSpellEntity.java
index ef93ff15..358d7b16 100644
--- a/src/main/java/com/minelittlepony/unicopia/entity/CastSpellEntity.java
+++ b/src/main/java/com/minelittlepony/unicopia/entity/CastSpellEntity.java
@@ -86,6 +86,11 @@ public class CastSpellEntity extends LightEmittingEntity implements Caster<Livin
         return Caster.of(getMaster()).map(Caster::getLevel).orElse(LEVELS);
     }
 
+    @Override
+    public LevelStore getCorruption() {
+        return Caster.of(getMaster()).map(Caster::getCorruption).orElse(LEVELS);
+    }
+
     @Override
     public Affinity getAffinity() {
         return getSpellSlot().get(true).map(Spell::getAffinity).orElse(Affinity.NEUTRAL);
diff --git a/src/main/java/com/minelittlepony/unicopia/entity/Creature.java b/src/main/java/com/minelittlepony/unicopia/entity/Creature.java
index fa55106e..39d5e0e3 100644
--- a/src/main/java/com/minelittlepony/unicopia/entity/Creature.java
+++ b/src/main/java/com/minelittlepony/unicopia/entity/Creature.java
@@ -9,8 +9,7 @@ import org.jetbrains.annotations.Nullable;
 import com.minelittlepony.unicopia.Affinity;
 import com.minelittlepony.unicopia.Race;
 import com.minelittlepony.unicopia.WeaklyOwned;
-import com.minelittlepony.unicopia.ability.magic.Affine;
-import com.minelittlepony.unicopia.ability.magic.Levelled;
+import com.minelittlepony.unicopia.ability.magic.*;
 import com.minelittlepony.unicopia.ability.magic.spell.Spell;
 import com.minelittlepony.unicopia.ability.magic.spell.effect.TargetSelecter;
 import com.minelittlepony.unicopia.entity.ai.BreakHeartGoal;
@@ -209,6 +208,11 @@ public class Creature extends Living<LivingEntity> implements WeaklyOwned<Living
         return LEVELS;
     }
 
+    @Override
+    public LevelStore getCorruption() {
+        return LEVELS;
+    }
+
     @Override
     public boolean subtractEnergyCost(double amount) {
         getMaster().damage(DamageSource.MAGIC, (int)amount/2);