Small Python 2.6 compatibility issue in test.py resolved