Minor tweak on example