TST fix broken doctest