Fix setup.py test