Updated example to reflect code change