Fix for plotting.py