fix test for Python 3