Fix py3 tests