Fix legend a little bit