fix failed test
Merge pull request #109 from pyp/fix_failed_test

fix failed test