I have been having some trouble recently recalling puzzle when she is in water. If she just goes in for a swim she has an urge to try and retrieve something even if i did not throw anything for her to retrieve. If i catch her before she goes in she will come back but its not so easy when she is swimming. If we are doing retrieves she is fine i have no trouble getting her out. I have tried walking away but she gets so engrossed in trying to find something to retrieve that she is not focused on me and does not notice i have walked off.