diff --git a/beets/importer.py b/beets/importer.py
index 385c59185..e97b0a75c 100644
--- a/beets/importer.py
+++ b/beets/importer.py
@@ -754,7 +754,7 @@ class ImportTask(BaseImportTask):
             self.record_replaced(lib)
             self.remove_replaced(lib)
             self.album = lib.add_album(self.imported_items())
-            if 'data_source' in self.imported_items()[0].keys():
+            if 'data_source' in self.imported_items()[0]:
                 self.album.data_source = self.imported_items()[0].data_source
             self.reimport_metadata(lib)