Fix a bug in the shader.