Fixed minor problem in code snippet