Fix example for api changes