FIX missing import in doctest