fix and add tests

テストを変更に追随させ、追加しました。