test.py fix