Fix clicking FluidDisplay not working with digital tank overflow mode (#1149)

2 files changed, 10 insertions, 1 deletions

diff --git a/src/main/java/gregtech/api/gui/GT_Container.java b/src/main/java/gregtech/api/gui/GT_Container.java
index 2b7959d6cb..843d34c1f3 100644
--- a/src/main/java/gregtech/api/gui/GT_Container.java
+++ b/src/main/java/gregtech/api/gui/GT_Container.java
@@ -691,7 +691,7 @@ public class GT_Container extends Container {
             tNewFillableStack.amount = tAmountTaken * tParallel;
             aFluidAccess.set(tNewFillableStack);
         } else {
-            aFluidAccess.get().amount += tAmountTaken * tParallel;
+            aFluidAccess.addAmount(tAmountTaken * tParallel);
         }
         tStackEmptied.stackSize = tParallel;
         replaceCursorItemStack(aPlayer, tStackEmptied);
diff --git a/src/main/java/gregtech/api/gui/GT_Container_BasicTank.java b/src/main/java/gregtech/api/gui/GT_Container_BasicTank.java
index 4d533f76c3..d294b60ca9 100644
--- a/src/main/java/gregtech/api/gui/GT_Container_BasicTank.java
+++ b/src/main/java/gregtech/api/gui/GT_Container_BasicTank.java
@@ -6,6 +6,7 @@ import gregtech.api.interfaces.IFluidAccess;
 import gregtech.api.interfaces.tileentity.IGregTechTileEntity;
 import gregtech.api.metatileentity.implementations.GT_MetaTileEntity_BasicTank;
 import gregtech.api.util.GT_Utility;
+import gregtech.common.tileentities.storage.GT_MetaTileEntity_DigitalTankBase;
 import net.minecraft.entity.player.EntityPlayer;
 import net.minecraft.entity.player.InventoryPlayer;
 import net.minecraft.inventory.ICrafting;
@@ -129,6 +130,14 @@ public class GT_Container_BasicTank extends GT_ContainerMetaTile_Machine {
             return mTank.getCapacity();
         }
 
+        @Override
+        public int getRealCapacity() {
+            if (mTank instanceof GT_MetaTileEntity_DigitalTankBase) {
+                return ((GT_MetaTileEntity_DigitalTankBase) mTank).getRealCapacity();
+            }
+            return IFluidAccess.super.getRealCapacity();
+        }
+
         static BasicTankFluidAccess from(GT_MetaTileEntity_BasicTank aTank, boolean aIsFillableStack) {
             return new BasicTankFluidAccess(aTank, aIsFillableStack);
         }