Fixing code snippet.