Update doctests that use