Small fix to the plot.