Update comment to match with code