Fix the test runner for Python