fix test for python 3