minor fix for Word2Vec test