Fix test for Python3