You may have done better by focussing on the rocks beforehand in the plane of the water or as mentioned, a smaller aperture for more depth of field. Also, manual focus would have done the trick.
The flash can also freeze the water mid-air allowing for a shaper image.
