Fixed import and doctest.