--- a/net/minecraft/world/entity/ai/behavior/warden/Digging.java
+++ b/net/minecraft/world/entity/ai/behavior/warden/Digging.java
@@ -39,7 +_,7 @@
     @Override
     protected void stop(ServerLevel level, E entity, long gameTime) {
         if (entity.getRemovalReason() == null) {
-            entity.remove(Entity.RemovalReason.DISCARDED);
+            entity.remove(Entity.RemovalReason.DISCARDED, org.bukkit.event.entity.EntityRemoveEvent.Cause.DESPAWN); // CraftBukkit - Add bukkit remove cause
         }
     }
 }
