Update example to work when #97 lands