Fix for test in Python 2.6