Fix tranformation issues with shaders