Fix broken test on python 2.6