FIX broken doctest

Sorry the last PR introduced a mistake.