fix test/test.py