Fix a comment to reflect correct output