Fix the doc blocks