Make doctest example a bit more compact.