Fix test for python 2.6