fix test.py reg