Fix a bug by assininge the shader uniforms