Quick fix for legend issue.