Fix test
Merge pull request #5 from tyabe/fix_test

Fix test