diff --git a/lua/entities/gmod_wire_expression2/core/custom/lifesupport.lua b/lua/entities/gmod_wire_expression2/core/custom/lifesupport.lua
index 5dd2f27..5db3010 100644
--- a/lua/entities/gmod_wire_expression2/core/custom/lifesupport.lua
+++ b/lua/entities/gmod_wire_expression2/core/custom/lifesupport.lua
@@ -144,7 +144,7 @@ end
 
 -- PUMP FUNCTIONS
 local function ls_ensure_valid_pump(ent)
-	return ent and IsValid(ent) and ent:GetClass() == "rd_pump"
+	return ent and IsValid(ent) and ent.IsPump
 end
 
 e2function number entity:lsPumpSetResourceAmount(string res, number amount)