Fix test suite in setup.py