Refactor code a little

trivialなリファクタリングです