Fix a comment in main