unit testを通るように修正