Fixed test_setup.py