Finally fixed the shader.