Fixed the key of the dictionary.

Fixed the key of the dictionary.