fix legend bug