Fixed problem with doctest