Fix Shader bug from refactoring