Fixed Python testing script