fix unit test
Merge pull request #25 from box/unittest

fix unit test