Fix test failure in python2.6