Fixed test.py