When ever I glide the elytra into the water while holding the space bar the elytra stays engaged on till I let go of space and hit the bottom of the ocean or reach land. you can also flip any direction by hitting the move buttons on your keyboard. The attached picture should give you a good idea of what's going on.