Fixed test for python2.6