Make doctest pass