Fix tests compatibility with Python 2.6