make it work on Python3