python: fix python state tracker after some latest changes