Minor bug fix in plotter