diff --git a/graphsage/utils.py b/graphsage/utils.py
index 4143d46..ff05072 100644
--- a/graphsage/utils.py
+++ b/graphsage/utils.py
@@ -8,7 +8,7 @@ import os
 
 import networkx as nx
 from networkx.readwrite import json_graph
-version_info = map(int, nx.__version__.split('.'))
+version_info = list(map(int, nx.__version__.split('.')))
 major = version_info[0]
 minor = version_info[1]
 assert (major <= 1) and (minor <= 11), "networkx major version > 1.11"