From 916913337b8b36150e2bbd1d55478f90874fc854 Mon Sep 17 00:00:00 2001 From: modmuss50 Date: Sat, 5 Nov 2016 10:43:00 +0000 Subject: [PATCH] Fix #789 --- .../java/techreborn/tiles/generator/TileGenerator.java | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/src/main/java/techreborn/tiles/generator/TileGenerator.java b/src/main/java/techreborn/tiles/generator/TileGenerator.java index 45bf1f0c0..66ec82f9b 100644 --- a/src/main/java/techreborn/tiles/generator/TileGenerator.java +++ b/src/main/java/techreborn/tiles/generator/TileGenerator.java @@ -2,9 +2,11 @@ package techreborn.tiles.generator; import net.minecraft.block.state.IBlockState; import net.minecraft.entity.player.EntityPlayer; +import net.minecraft.init.Items; import net.minecraft.item.ItemStack; import net.minecraft.tileentity.TileEntityFurnace; import net.minecraft.util.EnumFacing; +import net.minecraftforge.common.ForgeModContainer; import reborncore.api.power.EnumPowerTier; import reborncore.api.tile.IInventoryProvider; import reborncore.common.IWrenchable; @@ -56,7 +58,12 @@ public class TileGenerator extends TilePowerAcceptor implements IWrenchable, IIn updateState(); burnItem = getStackInSlot(fuelSlot); if (getStackInSlot(fuelSlot).stackSize == 1) { - setInventorySlotContents(fuelSlot, null); + if(getStackInSlot(fuelSlot).getItem() == Items.LAVA_BUCKET || getStackInSlot(fuelSlot).getItem() == ForgeModContainer.getInstance().universalBucket){ + setInventorySlotContents(fuelSlot, new ItemStack(Items.BUCKET)); + } else { + setInventorySlotContents(fuelSlot, null); + } + } else { decrStackSize(fuelSlot, 1); }