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

1 files changed, 18 insertions, 1 deletions

diff --git a/src/main/java/gregtech/common/tileentities/storage/GT_MetaTileEntity_DigitalTankBase.java b/src/main/java/gregtech/common/tileentities/storage/GT_MetaTileEntity_DigitalTankBase.java
index 77bf87dcc2..b8babfcea5 100644
--- a/src/main/java/gregtech/common/tileentities/storage/GT_MetaTileEntity_DigitalTankBase.java
+++ b/src/main/java/gregtech/common/tileentities/storage/GT_MetaTileEntity_DigitalTankBase.java
@@ -160,6 +160,23 @@ public abstract class GT_MetaTileEntity_DigitalTankBase extends GT_MetaTileEntit
         GT_Utility.sendChatToPlayer(aPlayer, StatCollector.translateToLocal(mVoidOverflow ? "GT5U.machines.voidoveflow.enabled" : "GT5U.machines.voidoveflow.disabled"));
     }
 
+    @Override
+    public FluidStack setFillableStack(FluidStack aFluid) {
+        mFluid = aFluid;
+        if (mFluid != null) {
+            mFluid.amount = Math.min(mFluid.amount, getRealCapacity());
+        }
+        return mFluid;
+    }
+
+    @Override
+    public FluidStack setDrainableStack(FluidStack aFluid) {
+        mFluid = aFluid;
+        if (mFluid != null) {
+            mFluid.amount = Math.min(mFluid.amount, getRealCapacity());
+        }
+        return mFluid;
+    }
 
     @Override
     public void onPreTick(IGregTechTileEntity aBaseMetaTileEntity, long aTick) {
@@ -258,7 +275,7 @@ public abstract class GT_MetaTileEntity_DigitalTankBase extends GT_MetaTileEntit
         return mVoidOverflow ? Integer.MAX_VALUE : getRealCapacity();
     }
 
-    private int getRealCapacity(){
+    public int getRealCapacity(){
         return commonSizeCompute(mTier);
     }