Plot legend fix