`word_index` is a `defaultdict` and `word` isn't one of the keys. Mutability can sometimes be a real pain when you're not interested in the absolutely quickest way (computationally) to run something. Moral of the story: RTFM ;)