fix bug in code example