Trivial fixes for doctests

fixes #2