Corrected another bug that made doctest fail