When you say you were not able to connect do you mean you waited in vain for the video feed to return or did it not obey your controller inputs? Sometimes it will have a crash of the Free Flight 6 app but there is no actual loss of control. If a drone is in contact with the controller it won't engage RTH because it is in contact with the controller... The solution to this is to reboot the app. Naturally, when your drone is over water this isn't the first thing that leaps to mind. If you are in clear line of sight you don't need the screen to fly, just fly it VLOS. If the controller has crashed for some reason reboot that as well, at ten metres away reestablishing contact should be straight forward.
I had a similar issue a few weeks ago when the drone lost signal for no obvious reason and it returned to home a little less precisely than I would like. I had no control and it was hovering over the edge of a very cold waist-deep lake. Fortunately, it was just within reach of my tall son who asked me whether he could catch it like a Spark. Yes, you can snatch it out of the air and turn it 90 or more degrees over and the motors will cut out. After this incident, I played with a few more scenarios and I have experimented with rebooting the app and the controller mid-flight, although so far only when it has been hovering low and nearby. To boost your confidence it might be a good idea to test an emergency scenario so you can see what happens if you reboot the controller and app while the drone is hovering a modest distance away, at a height that you could grab it out of the sky if it didn't respond.