Change to regular Word2Vec model as Doc2Vec seems to mess up stuff...