Fixing the word2vec dict