Fix example and test