diff --git a/model.py b/model.py
index d59f46b..d755edf 100644
--- a/model.py
+++ b/model.py
@@ -259,7 +259,7 @@ def create_gensim_files(self, sess, model_folder, ind2emoj, out_dim):
             f.write('\n')
         f.close()
 
-        e2v = gs.Word2Vec.load_word2vec_format(txt_path, binary=False)
+        e2v = gs.KeyedVectors.load_word2vec_format(txt_path, binary=False)
         e2v.save_word2vec_format(bin_path, binary=True)
 
         return e2v