Small fix in the script