Make test pass on Python 2.7