Fix Python 3 support in tests.