Small fix in script.