revert changes in plot.py