fix test.py