Fix error in code snippet