Fix exception
Try fix on doctest again