fix example

Updated so the example works.