Small fix to code to match its description.