Fix failing doctests