Fix compatibility with Python 2