Payne, some dogs will try to walk on their hind legs instead of swim. That leads to a head high flailing. Once they learn to get their head down and out front they get it together. Sounds like he likes the water so let him work it out at his pace. If he is retrieving good and likes birds, a frozen teal can work wonders, but I wouldn't sweat it. Don't force it and he will get it.
Smokey Bear---Lone Star State.