【Swift】UICollectionViewのセルを移動させる時に、セクションを跨げないようにする方法

extension ViewController: UICollectionViewDelegate {

func collectionView(_ collectionView: UICollectionView, targetIndexPathForMoveFromItemAt originalIndexPath: IndexPath, toProposedIndexPath proposedIndexPath: IndexPath) -> IndexPath {
  //移動元と移動先のセクション番号が異なる場合
  if originalIndexPath.section != proposedIndexPath.section {
    return originalIndexPath
  } else {
      //セクション番号が一致する場合
      return proposedIndexPath
    }
  }
}