When riding a non-flying, non-swimming steed, give a warning by making the steed stop and “shy away” before riding into water. Only allow this if you use the m command or the steed has already shied away once. But it’s not very fair or realistic to have the steed instadrown under you.