Update comment to reflect code changes