Fix test for Py3