Fixed doctest
Fixed doctest