Fix setup.py test script.