fix the run.py
fix the run.py