1 files changed, 4 insertions, 4 deletions
diff --git a/src/main/kotlin/gui/entity/ModifyCharged.kt b/src/main/kotlin/gui/entity/ModifyCharged.kt
index d22f6e3..23fd495 100644
--- a/src/main/kotlin/gui/entity/ModifyCharged.kt
+++ b/src/main/kotlin/gui/entity/ModifyCharged.kt
@@ -2,13 +2,13 @@
 package moe.nea.firmament.gui.entity
 
 import com.google.gson.JsonObject
-import net.minecraft.entity.LivingEntity
-import net.minecraft.entity.mob.CreeperEntity
+import net.minecraft.world.entity.LivingEntity
+import net.minecraft.world.entity.monster.Creeper
 
 object ModifyCharged : EntityModifier {
     override fun apply(entity: LivingEntity, info: JsonObject): LivingEntity {
-        require(entity is CreeperEntity)
-        entity.dataTracker.set(CreeperEntity.CHARGED, true)
+        require(entity is Creeper)
+        entity.entityData.set(Creeper.DATA_IS_POWERED, true)
         return entity
     }
 }