Fixing setup.py test running