Fix to the script