minor change to run.py