Fixed custom AI goals on 1.20

Auxilor · Nov 7, 2023 · e306190 · e306190
1 parent 3e0a8b1
commit e306190
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.../nms-common/src/main/kotlin/com/willfp/eco/internal/spigot/proxy/common/ai/CustomGoals.kt b/.../nms-common/src/main/kotlin/com/willfp/eco/internal/spigot/proxy/common/ai/CustomGoals.kt
@@ -2,6 +2,7 @@ package com.willfp.eco.internal.spigot.proxy.common.ai
 
 import com.willfp.eco.core.entities.ai.CustomGoal
 import com.willfp.eco.core.entities.ai.GoalFlag
+import com.willfp.eco.internal.spigot.proxy.common.toBukkitEntity
 import net.minecraft.world.entity.PathfinderMob
 import net.minecraft.world.entity.ai.goal.Goal
 import java.util.EnumSet
@@ -51,7 +52,7 @@ private class NMSCustomGoal<T : org.bukkit.entity.Mob>(
 ) : Goal() {
     init {
         @Suppress("UNCHECKED_CAST")
-        customEntityGoal.initialize(entity.bukkitMob as T)
+        customEntityGoal.initialize(entity.toBukkitEntity() as T)
         this.setFlags(EnumSet.copyOf(customEntityGoal.flags.toNMSFlags()))
     }