diff --git a/src/main/java/com/minelittlepony/unicopia/mixin/MixinMobEntity.java b/src/main/java/com/minelittlepony/unicopia/mixin/MixinMobEntity.java index f86144dc..23088e21 100644 --- a/src/main/java/com/minelittlepony/unicopia/mixin/MixinMobEntity.java +++ b/src/main/java/com/minelittlepony/unicopia/mixin/MixinMobEntity.java @@ -9,9 +9,12 @@ import org.spongepowered.asm.mixin.injection.callback.CallbackInfo; import com.minelittlepony.unicopia.entity.Creature; import com.minelittlepony.unicopia.entity.PonyContainer; import com.minelittlepony.unicopia.entity.Equine; + +import net.minecraft.entity.EntityType; import net.minecraft.entity.LivingEntity; import net.minecraft.entity.ai.goal.GoalSelector; import net.minecraft.entity.mob.MobEntity; +import net.minecraft.world.World; @Mixin(MobEntity.class) abstract class MixinMobEntity extends LivingEntity implements PonyContainer> { @@ -22,8 +25,8 @@ abstract class MixinMobEntity extends LivingEntity implements PonyContainer entityType, World world, CallbackInfo info) { ((Creature)get()).initAi(goalSelector, targetSelector); } }