Fixing bug in code example