fix to example code