Update run.py

Ubuntu fix