Minor fix in run.py