Fix Dataset docstring