fix the plotting