Fix to work on python3