Fix some docstring