Removing gensim warning