slight tweak to example