Fixed issue in py 3 tests.