diff --git a/src/main/java/com/gtnewhorizons/postea/utility/ChunkFixerUtility.java b/src/main/java/com/gtnewhorizons/postea/utility/ChunkFixerUtility.java index caa6132..0d2745c 100644 --- a/src/main/java/com/gtnewhorizons/postea/utility/ChunkFixerUtility.java +++ b/src/main/java/com/gtnewhorizons/postea/utility/ChunkFixerUtility.java @@ -130,11 +130,11 @@ public class ChunkFixerUtility { BlockInfo blockInfo = transformationFunction.apply(tileEntity, world, chunk); if (blockInfo == null) continue; - if (blockInfo.tileTransformer == null) { - tileEntities.removeTag(i--); + NBTTagCompound newTag; + if (blockInfo.tileTransformer != null && (newTag = blockInfo.tileTransformer.apply(tileEntity)) != null) { + ((IMixinNBTTagList) tileEntities).Postea$replaceCompoundTagAt(i, newTag); } else { - ((IMixinNBTTagList) tileEntities) - .Postea$replaceCompoundTagAt(i, blockInfo.tileTransformer.apply(tileEntity)); + tileEntities.removeTag(i--); } conversionInfo.add(new ConversionInfo(x, y, z, blockInfo));