Fix example

Closes #2