--- a/net/minecraft/world/level/entity/EntityAccess.java
+++ b/net/minecraft/world/level/entity/EntityAccess.java
@@ -6,6 +6,10 @@
 import net.minecraft.world.entity.Entity;
 import net.minecraft.world.phys.AxisAlignedBB;
 
+// CraftBukkit start
+import org.bukkit.event.entity.EntityRemoveEvent;
+// CraftBukkit end
+
 public interface EntityAccess {
 
     int getId();
@@ -24,6 +28,12 @@
 
     void setRemoved(Entity.RemovalReason entity_removalreason);
 
+    // CraftBukkit start - add Bukkit remove cause
+    default void setRemoved(Entity.RemovalReason entity_removalreason, EntityRemoveEvent.Cause cause) {
+        setRemoved(entity_removalreason);
+    }
+    // CraftBukkit end
+
     boolean shouldBeSaved();
 
     boolean isAlwaysTicking();
