Simplify the shader as previously thanks to @zz85