Fixed test breaking on python 2.6