Updated example to be consistent with other example