fix failed doctest
Merge pull request #9 from ma2gedev/fix_doctest

fix failed doctest