Fix test for python26