コードを一部修正