Fixed test runner, wasn't working with `python setup.py test`