small fix in code snippet