Change comment to match code