Minor tweak to a code example