Hi, Thanks for your feedback!
We have just release a new update for the game, and it adds a new Joystick System, and you can choose it from the Settings Menu. Just open the Pause Menu during an underwater level, then go to Settings and change Joystick!
We hope you like this new movement system, please let us know your feedback!