トレイトを使用したテストを修正