fix test in py3