Fix test in setup.py