Fix test for python2/3 compat