テストケースに合うように修正