Cole back to running in the water