Corrected a change in test.py