Fix test for Py3 support.