Fix missed comment change