Fixed bugs in shaders