Shader bug fix