Fixed broken example after API change

Addresses #9