Fix example code

Fix gh-11