--- ../src-base/minecraft/net/minecraft/world/gen/feature/WorldGenDungeons.java
+++ ../src-work/minecraft/net/minecraft/world/gen/feature/WorldGenDungeons.java
@@ -16,6 +16,9 @@
 import net.minecraft.world.World;
 import org.apache.logging.log4j.LogManager;
 import org.apache.logging.log4j.Logger;
+import net.minecraftforge.common.ChestGenHooks;
+import net.minecraftforge.common.DungeonHooks;
+import static net.minecraftforge.common.ChestGenHooks.DUNGEON_CHEST;
 
 public class WorldGenDungeons extends WorldGenerator
 {
@@ -140,12 +143,11 @@
                                 if (k2 == 1)
                                 {
                                     p_180709_1_.func_180501_a(blockpos2, Blocks.field_150486_ae.func_176458_f(p_180709_1_, blockpos2, Blocks.field_150486_ae.func_176223_P()), 2);
-                                    List list = WeightedRandomChestContent.func_177629_a(field_175917_c, new WeightedRandomChestContent[] {Items.field_151134_bR.func_92114_b(p_180709_2_)});
                                     TileEntity tileentity1 = p_180709_1_.func_175625_s(blockpos2);
 
                                     if (tileentity1 instanceof TileEntityChest)
                                     {
-                                        WeightedRandomChestContent.func_177630_a(p_180709_2_, list, (TileEntityChest)tileentity1, 8);
+                                        WeightedRandomChestContent.func_177630_a(p_180709_2_, ChestGenHooks.getItems(DUNGEON_CHEST, p_180709_2_), (TileEntityChest)tileentity1, ChestGenHooks.getCount(DUNGEON_CHEST, p_180709_2_));
                                     }
 
                                     break label100;
@@ -184,6 +186,12 @@
 
     private String func_76543_b(Random p_76543_1_)
     {
-        return field_175916_b[p_76543_1_.nextInt(field_175916_b.length)];
+        return DungeonHooks.getRandomDungeonMob(p_76543_1_);
     }
+
+    static
+    {
+        ChestGenHooks.init(DUNGEON_CHEST, field_175917_c, 8, 8);
+        ChestGenHooks.addItem(DUNGEON_CHEST, new WeightedRandomChestContent(new net.minecraft.item.ItemStack(Items.field_151134_bR, 1, 0), 1, 1, 1));
+    }
 }
