Endermen teleport when they touch water, but not lava

This is silly. When Endermen come into contact with water they immediately teleport away from it. But, when they come into contact with lava, they don't, and will burn up and die. Fixing this would break some mob traps, but people could just use pistons instead so it's no big deal.