Undead mobs sink in lava in stead of swminimg

Undead mobs don't swim in lava as they used to in 1.12 and earlier. They just sink.

This makes it really easy to get rid of zombie pigman in the nether for example.

However in the case of zombie pigman. They just stay there, causing some lagg when a pile of them builds up.