Fixing a test on python 2.6