Fix the python test run on travis.