Updated but still getting a KeyError.

Fixed the previous error but still getting a KeyError nonetheless.