Fix test shaders