FIX broken doctest