Fixed `python setup.py test`