Fixing tests on python 2.6