Fix test for Python 2.6