Fix test cases for doctest.
2 files changed