the implementation on the python side now seems to work just fine.