fix broken doctests