Minor tweak to example