Fix example so that it actually works