fixed test.py