Fixing code comment