fix test for Python3