fix test.py
fix test