From d2d203f7727c8bfee29463ca49d8161a53d8d2fb Mon Sep 17 00:00:00 2001
From: mezz <tehgeek@gmail.com>
Date: Wed, 21 Oct 2015 14:15:37 -0700
Subject: [PATCH] Fix machines won't drain fluid containers that hold less than
 one bucket

---
 src/main/java/forestry/core/fluids/FluidHelper.java | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/src/main/java/forestry/core/fluids/FluidHelper.java b/src/main/java/forestry/core/fluids/FluidHelper.java
index e064ae96d..83dd4edf5 100644
--- a/src/main/java/forestry/core/fluids/FluidHelper.java
+++ b/src/main/java/forestry/core/fluids/FluidHelper.java
@@ -233,11 +233,14 @@ public final class FluidHelper {
 			IFluidContainerItem containerItem = (IFluidContainerItem) item;
 			containerItem.drain(drained, drainAmount, true);
 			return drained;
-		} else if (FluidContainerRegistry.isContainer(container)) {
-			if (drainAmount < Constants.BUCKET_VOLUME) {
-				return container;
+		} else {
+			int capacity = FluidContainerRegistry.getContainerCapacity(container);
+			if (capacity > 0) {
+				if (drainAmount < capacity) {
+					return container;
+				}
+				return FluidContainerRegistry.drainFluidContainer(container);
 			}
-			return FluidContainerRegistry.drainFluidContainer(container);
 		}
 
 		return null;