--- a/net/minecraft/world/level/entity/EntityAccess.java
+++ b/net/minecraft/world/level/entity/EntityAccess.java
@@ -5,8 +5,12 @@
 import net.minecraft.core.BlockPos;
 import net.minecraft.world.entity.Entity;
 import net.minecraft.world.phys.AABB;
+// CraftBukkit start
+import org.bukkit.event.entity.EntityRemoveEvent;
+// CraftBukkit end
 
 public interface EntityAccess {
+
     int getId();
 
     UUID getUUID();
@@ -23,6 +27,12 @@
 
     void setRemoved(Entity.RemovalReason reason);
 
+    // CraftBukkit start - add Bukkit remove cause
+    default void setRemoved(Entity.RemovalReason entity_removalreason, EntityRemoveEvent.Cause cause) {
+        this.setRemoved(entity_removalreason);
+    }
+    // CraftBukkit end
+
     boolean shouldBeSaved();
 
     boolean isAlwaysTicking();
