minor changes in test.py