fix import in doctest