Fix tests with Python 2.6