Fix example

The example didn't build but will with these changes.