python script fix