テストを通るように修正