fix broken tests

テスト壊れてたのを直しました。プロダクションコードの一部がテスト通らないことが判明しました。