Fix test for py3