Swiftで書きなおしたクラスに一部不具合があった部分を修正