Fixed doctest import