Partially fix the rendering of water (#4)