Change to example code in line with bug fix for issue 10