fix test cases fail on remove()
Merge pull request #5 from fengmk2/fix-test

fix test cases fail on remove()