Undead mobs don't swim up in water

When zombies are in water they don’t swim to the surface like other mobs and they get stuck there. I know they can turn into a drowned but this is not intended because zombies are meant to swim.