Make player/entity immune to lava contact damage (not the being set on fire or burning ticks)

What I'm mainly looking for is whether there is a simple entity tag or flag that handles this?
I would still like them to be set on fire and take burning damage as normal if possible (though if it also prevents this, doesn't really matter to me)