If your baby goes into water too deep for it walk it will fly and not land. The only option to get it back on the ground is to use a chain bola otherwise it will likely starve unless you can manage to get into its inventory. Knocking it out doesn't work and neither does a whip
I got a level 150 lightning egg and I was so excited to hatch the egg...that when I went to put the egg down I accidentally ate it instead. :( ps4 users: R2 ISN'T GONNA DROP THE EGG YOU'RE GONNA EAT IT LIKE ANY OTHER EGG. REMEMBER THIS. LOGIC BEFORE EXCITMENT.