Fix example

Fixes #523