fix plotting in run_kmod.py