minor changes on python side