Unbreak tests on Python 2.x