diff --git a/data/ami/utils.py b/data/ami/utils.py
index cb671070..2ff5fac9 100644
--- a/data/ami/utils.py
+++ b/data/ami/utils.py
@@ -108,7 +108,7 @@ def get_fid2length(train_file):
 
 def full_records(speakers, fid2length, subset_name=None):
     all_records = []
-    speakers = {(speaker.id, speaker) for speaker in speakers}
+    speakers = {speaker.id: speaker for speaker in speakers}
 
     for fid, length in fid2length:
         speaker = fid.split("_")[2]