This should fix the tests on Python 3.x