Hey thanks to you and your lovely kiddo for playing! <3
I'll give you a hint :D
The puzzle you were stuck on is not a timing puzzle. You already discovered your two possible actions, opening the gate with the robot upstairs and switching the water on/off. This is everything you need and once you have the solution you can take all the time in the world to swim to the exit <3