align with test.py