Fix error in code sample