Repaired Python test script.