Switched out the list comprehension for a dict comprehension.