Fixed small bug in plot.