Correct comment to match class name