Fixing test on python 2.6