fix plot.py