Fix py3 test failure