Fix the script again