fix example import in docstring