Fix water distortion in shader