Fix comments for doc generation