Add import statement for doctest.