Fix example. Closes #2