Small fix to make the script work.