Fix mistake in test_word2vec.py