fix test in Py3.1