Correct imports in docstrings