py3 unit test fix