Make this work on Python 2