Updated example to work with last changes.