correct the python script;