Make it work on Python 3