Fixes to shaders