Fix test in python 2.6