Revert "Updated but still getting a KeyError."

This reverts commit 0e3c439bdbf5198fc1683645f88469c79aa906f2.