Fixed the primary key extraction list comprehension.