Another fix for legend