Fix comment mismatch