Fix broken doctests