Having done whitewater kayaking for a long time I cannot stress this enough. You need to practice being under water with being denied surfacing even if you strongly want to.

Once you are calm, you can orient yourself by observing air bubbles for example, even in turbulent water.