Fix example code slightly