BUG: Fix doctest error in nosetests