Fix issue with running on python 2.6