Fix example implementation