test.py fixes