Python 2.6 test fix