Updated comment to reflect code change