Since there's a lot of problems with other mods damaging per tick and whatnot, what if there was a config option where you stored the last type of damage and entity that dealt said damage to you? Right before being damaged again, you only lose damage immunity if the damage source is different. It would reduce problems with wither, lava, etc. instantly gibbing you, while still keeping the danger of being ganged upon.