Fix example code in doc