fix bug in plot_utils.py