fix test in module
Merge pull request #30 from fengmk2/fix-test

fix test in module