[FIX] Update the script