Fix test in python 3.x