update tests
Merge pull request #18 from Hilaryous/test_fix

update tests