fixup up doctest after refactor