Fixing the script ...