diff --git a/picard/ui/searchdialog.py b/picard/ui/searchdialog.py
index 0cf29f48d..b9663b384 100644
--- a/picard/ui/searchdialog.py
+++ b/picard/ui/searchdialog.py
@@ -639,7 +639,7 @@ class AlbumSearchDialog(SearchDialog):
         if not self.table:
             return
 
-        cover_cell = self.table.cellWidget(row, len(self.table_headers) - 1)
+        cover_cell = self.table.cellWidget(row, self.colpos('cover'))
 
         if error:
             cover_cell.not_found()
@@ -677,7 +677,7 @@ class AlbumSearchDialog(SearchDialog):
         Args:
             row -- Album's row in results table
         """
-        cover_cell = self.table.cellWidget(row, len(self.table_headers) - 1)
+        cover_cell = self.table.cellWidget(row, self.colpos('cover'))
 
         if error:
             cover_cell.not_found()