Fix tests on python-2.6