fix example code for docs