Update run_test.py

Seems it works and clean.