If one of them swims away just look at your tame list, Wait in the water, and boom they’re there (you’ll have to tp them to you)