When I go lap swimming, and feel my body get caught up in the water, I look down at the long blue line at the bottom of the pool, and let it be my guide.  I fix my direction on that blue line, and concentrate on it as I feel my body weaken, until I get back to where I started and complete my lap.