Small fix in the script.