Modify example to fit new tests

- Fixes #21