Fix tests on Python 3.x