Fix the python script that runs the tests.