fixing doctest imports