fix test for python3