fix for doctest failure