Fix test in python 3