put doctests imports where they should be