--- a/net/minecraft/world/level/block/entity/TestInstanceBlockEntity.java
+++ b/net/minecraft/world/level/block/entity/TestInstanceBlockEntity.java
@@ -154,6 +_,7 @@
 
     @Override
     protected void loadAdditional(ValueInput input) {
+        super.loadAdditional(input); // Paper - load the PDC
         input.read("data", TestInstanceBlockEntity.Data.CODEC).ifPresent(this::set);
     }
 
@@ -317,7 +_,7 @@
     }
 
     private void removeEntities() {
-        this.level.getEntities(null, this.getStructureBounds()).stream().filter(entity -> !(entity instanceof Player)).forEach(Entity::discard);
+        this.level.getEntities(null, this.getStructureBounds()).stream().filter(entity -> !(entity instanceof Player)).forEach((entity) -> entity.discard(org.bukkit.event.entity.EntityRemoveEvent.Cause.DISCARD)); // Paper
     }
 
     private void forceLoadChunks() {
