small run.py change