Fix typo in test.py