Update comment to match code