Fix py3 test issue