Fix 'failing' doctest