Add correct color to water shader