Fix test on python 3