Fix a small problem in code example