Small cleanup in comments