Prevent Mobs dismounting when underwater

Literally just the title. I want to make a mob not dismount the mob it's riding when it goes underwater.