Fixed doctest