Fixed code example in comments