Fix test for python2.6