Fixing problems with legend updates