fixes in test.py