Yet another run.py change