Make the python binding work with the new version