Fix Python 3 tests