Fix test
Merge pull request #75 from chen3feng/develop

Fix test