Update unit test
Merge pull request #2 from alicoding/testUpdate

Update unit test