Fix test error on python2.6