fix legend on plots for paper