forgot some change to the python module