Make it work on python3