diff --git a/Persephone/Controllers/AlbumViewController+NSCollectionViewDelegate.swift b/Persephone/Controllers/AlbumViewController+NSCollectionViewDelegate.swift
index 2a52547..16b6a69 100644
--- a/Persephone/Controllers/AlbumViewController+NSCollectionViewDelegate.swift
+++ b/Persephone/Controllers/AlbumViewController+NSCollectionViewDelegate.swift
@@ -14,8 +14,8 @@ extension AlbumViewController: NSCollectionViewDelegate {
     collectionView.setDraggingSourceOperationMask(.copy, forLocal: true)
   }
 
-  func collectionView(_ collectionView: NSCollectionView, pasteboardWriterForItemAt index: Int) -> NSPasteboardWriting? {
-    let album = dataSource.albums[index]
+  func collectionView(_ collectionView: NSCollectionView, pasteboardWriterForItemAt indexPath: IndexPath) -> NSPasteboardWriting? {
+    let album = dataSource.albums[indexPath.item]
 
     return NSPasteboardItem(
       draggedAlbum: DraggedAlbum(
@@ -26,7 +26,7 @@ extension AlbumViewController: NSCollectionViewDelegate {
     )
   }
 
-  func collectionView(_ collectionView: NSCollectionView, canDragItemsAt indexes: IndexSet, with event: NSEvent) -> Bool {
+  func collectionView(_ collectionView: NSCollectionView, canDragItemsAt indexPaths: Set<IndexPath>, with event: NSEvent) -> Bool {
     return true
   }
 }