Fix test for Py 3