Fixing setup.py to reflect the test_data changes