Fix wardens not having their dig cooldown set properly

CitizensDev · Oct 28, 2022 · 45114dc · 45114dc
1 parent d532dae
commit 45114dc
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/v1_19_R1/src/main/java/net/citizensnpcs/nms/v1_19_R1/entity/WardenController.java b/v1_19_R1/src/main/java/net/citizensnpcs/nms/v1_19_R1/entity/WardenController.java
@@ -18,9 +18,11 @@
 import net.minecraft.nbt.CompoundTag;
 import net.minecraft.server.level.ServerLevel;
 import net.minecraft.sounds.SoundEvent;
+import net.minecraft.util.Unit;
 import net.minecraft.world.damagesource.DamageSource;
 import net.minecraft.world.entity.Entity;
 import net.minecraft.world.entity.EntityType;
+import net.minecraft.world.entity.ai.memory.MemoryModuleType;
 import net.minecraft.world.entity.monster.warden.Warden;
 import net.minecraft.world.entity.vehicle.AbstractMinecart;
 import net.minecraft.world.entity.vehicle.Boat;
@@ -48,6 +50,7 @@ public EntityWardenNPC(EntityType<? extends Warden> types, Level level) {
         public EntityWardenNPC(EntityType<? extends Warden> types, Level level, NPC npc) {
             super(types, level);
             this.npc = (CitizensNPC) npc;
+            this.getBrain().setMemoryWithExpiry(MemoryModuleType.DIG_COOLDOWN, Unit.INSTANCE, 1200L);
         }
 
         @Override
@@ -87,6 +90,7 @@ public void customServerAiStep() {
                 return;
             }
             NMSImpl.updateMinecraftAIState(npc, this);
+
             if (npc.useMinecraftAI()) {
                 super.customServerAiStep();
             } else {