Fix test for python 3.