diff --git a/src/main/java/tconstruct/smeltery/logic/LavaTankLogic.java b/src/main/java/tconstruct/smeltery/logic/LavaTankLogic.java
index 8d6ea0e5..b36b8462 100644
--- a/src/main/java/tconstruct/smeltery/logic/LavaTankLogic.java
+++ b/src/main/java/tconstruct/smeltery/logic/LavaTankLogic.java
@@ -101,12 +101,7 @@ public class LavaTankLogic extends MultiServantLogic implements IFluidHandler
     {
         if (containsFluid())
         {
-            int id = tank.getFluid().fluidID;
-            if (id < 4096)
-            {
-                Block block = Block.getBlockById(id);
-                return block.getLightValue();
-            }
+            return tank.getFluid().getFluid().getLuminosity();
         }
         return 0;
     }