minor change in test.py