From 527e6301c441e7113b6cd217cebc6358f4b59f40 Mon Sep 17 00:00:00 2001
From: MelanX <info@melanx.de>
Date: Tue, 23 Dec 2025 13:44:29 +0100
Subject: [PATCH] Fix crash for blocks that use copy block entity data function

---
 .../moddingx/libx/impl/loot/CopyBlockEntityDataFunction.java   | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/main/java/org/moddingx/libx/impl/loot/CopyBlockEntityDataFunction.java b/src/main/java/org/moddingx/libx/impl/loot/CopyBlockEntityDataFunction.java
index 78fd27ea..56099860 100644
--- a/src/main/java/org/moddingx/libx/impl/loot/CopyBlockEntityDataFunction.java
+++ b/src/main/java/org/moddingx/libx/impl/loot/CopyBlockEntityDataFunction.java
@@ -13,6 +13,7 @@
 import net.minecraft.world.item.component.CustomData;
 import net.minecraft.world.level.block.Block;
 import net.minecraft.world.level.block.entity.BlockEntity;
+import net.minecraft.world.level.block.entity.BlockEntityType;
 import net.minecraft.world.level.storage.loot.LootContext;
 import net.minecraft.world.level.storage.loot.functions.LootItemConditionalFunction;
 import net.minecraft.world.level.storage.loot.functions.LootItemFunctionType;
@@ -22,6 +23,7 @@
 
 import javax.annotation.Nonnull;
 import java.util.List;
+import java.util.Objects;
 import java.util.Set;
 
 public class CopyBlockEntityDataFunction extends LootItemConditionalFunction {
@@ -65,6 +67,7 @@ protected ItemStack run(@Nonnull ItemStack stack, @Nonnull LootContext context)
                         data = data.update(nbt -> nbt.put(tagName, tag.copy()));
                     }
                 }
+                data = data.update(nbt -> nbt.putString("id", Objects.requireNonNull(BlockEntityType.getKey(blockEntity.getType())).toString()));
                 return data;
             });
         }