Fixed comment to reflect code change