Zombies stop chasing on water

escape from a zombie going to water. He will stop walking when his head is under water. He drowns and "Drowned" continues the chase. Make zombie continue walking till they drown since the time they wait you could already build you a refuge under water. At least what I know is that they wont go any block deeper to chase you.