diff --git a/node2vec/__init__.py b/node2vec/__init__.py
index 3748862..93d47d6 100644
--- a/node2vec/__init__.py
+++ b/node2vec/__init__.py
@@ -1,4 +1,4 @@
 from . import edges
 from .node2vec import Node2Vec
 
-__version__ = '0.4.1'
+__version__ = '0.4.2'
diff --git a/node2vec/node2vec.py b/node2vec/node2vec.py
index 1ef1693..c44e1de 100644
--- a/node2vec/node2vec.py
+++ b/node2vec/node2vec.py
@@ -1,10 +1,11 @@
-import random
 import os
+import random
 from collections import defaultdict
 
-import numpy as np
-import networkx as nx
 import gensim
+import networkx as nx
+import numpy as np
+import pkg_resources
 from joblib import Parallel, delayed
 from tqdm.auto import tqdm
 
@@ -166,7 +167,8 @@ def _generate_walks(self) -> list:
     def fit(self, **skip_gram_params) -> gensim.models.Word2Vec:
         """
         Creates the embeddings using gensim's Word2Vec.
-        :param skip_gram_params: Parameteres for gensim.models.Word2Vec - do not supply 'size' it is taken from the Node2Vec 'dimensions' parameter
+        :param skip_gram_params: Parameters for gensim.models.Word2Vec - do not supply 'size' / 'vector_size' it is
+            taken from the Node2Vec 'dimensions' parameter
         :type skip_gram_params: dict
         :return: A gensim word2vec model
         """
@@ -174,8 +176,11 @@ def fit(self, **skip_gram_params) -> gensim.models.Word2Vec:
         if 'workers' not in skip_gram_params:
             skip_gram_params['workers'] = self.workers
 
-        if 'size' not in skip_gram_params:
-            skip_gram_params['size'] = self.dimensions
+        # Figure out gensim version, naming of output dimensions changed from size to vector_size in v4.0.0
+        gensim_version = pkg_resources.get_distribution("gensim").version
+        size = 'size' if gensim_version < '4.0.0' else 'vector_size'
+        if size not in skip_gram_params:
+            skip_gram_params[size] = self.dimensions
 
         if 'sg' not in skip_gram_params:
             skip_gram_params['sg'] = 1
diff --git a/setup.py b/setup.py
index df4868d..ddcf802 100644
--- a/setup.py
+++ b/setup.py
@@ -3,7 +3,7 @@
 setup(
     name='node2vec',
     packages=['node2vec'],
-    version='0.4.1',
+    version='0.4.2',
     description='Implementation of the node2vec algorithm.',
     author='Elior Cohen',
     author_email='elior.cohen.p@gmail.com',