1 files changed, 2 insertions, 1 deletions
diff --git a/simulation/main.py b/simulation/main.py
index decdf8f..1c0b3e8 100644
--- a/simulation/main.py
+++ b/simulation/main.py
@@ -7,6 +7,7 @@ from scipy.optimize import minimize
 import matplotlib.pyplot as plt
 import seaborn
 from random import random, randint
+from six.moves import range
 
 seaborn.set_style("white")
 
@@ -39,7 +40,7 @@ def uniform_source(graph, *args, **kwargs):
 
 
 def simulate_cascades(n, graph, source=uniform_source):
-    for t in xrange(n):
+    for t in range(n):
         x0 = source(graph, t)
         yield simulate_cascade(x0, graph)