Fixing some doctest errors