More morbing

1 files changed, 4 insertions, 1 deletions

diff --git a/src/main/java/moe/nea/morbing/BloodFluid.java b/src/main/java/moe/nea/morbing/BloodFluid.java
index bbda497..4583967 100644
--- a/src/main/java/moe/nea/morbing/BloodFluid.java
+++ b/src/main/java/moe/nea/morbing/BloodFluid.java
@@ -74,7 +74,7 @@ public abstract class BloodFluid extends FlowableFluid {
         BlockPos abovePos = pos.up();
         BlockState aboveState = world.getBlockState(abovePos);
         FluidState aboveFluid = aboveState.getFluidState();
-        if (state.getFluidState().getFluid() != this && !aboveFluid.isEmpty() && matchesType(aboveFluid.getFluid())) {
+        if (state.getFluidState().isEmpty() && !aboveFluid.isEmpty() && matchesType(aboveFluid.getFluid())) {
             return getFlowing(aboveFluid.getLevel(), true);
         }
         if (state.getFluidState().getFluid() == getFlowing()) {
@@ -86,6 +86,9 @@ public abstract class BloodFluid extends FlowableFluid {
                 return getFlowing(1, false);
             }
         }
+        if (state.getFluidState().getFluid() == getStill()) {
+            return getFlowing(1, false);
+        }
         return state.getFluidState();
     }