--- a/net/minecraft/gametest/framework/GameTestInfo.java
+++ b/net/minecraft/gametest/framework/GameTestInfo.java
@@ -241,7 +_,7 @@
             AABB structureBounds = this.getStructureBounds();
             List<Entity> entitiesOfClass = this.getLevel()
                 .getEntitiesOfClass(Entity.class, structureBounds.inflate(1.0), entity -> !(entity instanceof Player));
-            entitiesOfClass.forEach(entity -> entity.remove(Entity.RemovalReason.DISCARDED));
+            entitiesOfClass.forEach(entity -> entity.remove(Entity.RemovalReason.DISCARDED, org.bukkit.event.entity.EntityRemoveEvent.Cause.DISCARD)); // Paper
         }
     }