fix test in python 3.x