diff --git a/src/main.py b/src/main.py
index 82ac735..ea08532 100644
--- a/src/main.py
+++ b/src/main.py
@@ -85,7 +85,7 @@ def learn_embeddings(walks):
 	'''
 	walks = [map(str, walk) for walk in walks]
 	model = Word2Vec(walks, size=args.dimensions, window=args.window_size, min_count=0, sg=1, workers=args.workers, iter=args.iter)
-	model.save_word2vec_format(args.output)
+	model.wv.save_word2vec_format(args.output)
 	
 	return