Fix run.py