Fix unit test
Merge pull request #33 from ogail/ut

Fix unit test