Py3 test fix