Fix test on py3