Firstly stay on edge longer. I know you're working on them one wake, so instead of cutting and then flattening out/coasting in, shorten up your approach and ride on your edge all the way up the wake.
Secondly, head position. Keep your head up, look down, go down. You're spotting the water while your back is to the boat and getting lost as you come around. Try to keep your head up and looking in the direction of your spin. As you near completing your rotation be spotting the water in your direction of travel.