Fixing the shader so it works now.