Update the second code example

Update the second code example to reflect the last change in the first example.