diff --git a/paper-server/src/main/java/org/bukkit/craftbukkit/generator/CustomChunkGenerator.java b/paper-server/src/main/java/org/bukkit/craftbukkit/generator/CustomChunkGenerator.java
index aa5d61f8bf..19d25ebea9 100644
--- a/paper-server/src/main/java/org/bukkit/craftbukkit/generator/CustomChunkGenerator.java
+++ b/paper-server/src/main/java/org/bukkit/craftbukkit/generator/CustomChunkGenerator.java
@@ -154,7 +154,7 @@ public class CustomChunkGenerator extends InternalChunkGenerator {
                 IBlockData block = craftData.getTypeId(tx, ty, tz);
 
                 if (block.isTileEntity()) {
-                    TileEntity tile = ((ITileEntity) block).createTile(new BlockPosition((x << 4) + tx, ty, (z << 4) + tz), block);
+                    TileEntity tile = ((ITileEntity) block.getBlock()).createTile(new BlockPosition((x << 4) + tx, ty, (z << 4) + tz), block);
                     ichunkaccess.setTileEntity(tile);
                 }
             }