Fix the doc comment