Fix example in code documentation