Fix test running through ./setup.py test