MAINT fix in docstring