diff --git a/setup.py b/setup.py
index 33bda862497..7aad7ceedfe 100644
--- a/setup.py
+++ b/setup.py
@@ -7,6 +7,7 @@
 
 requires = [
     'tornado >= 4.2, < 5.0.0',
+    'networkx',
     'numpy',
     'pandas',
     'tqdm',
diff --git a/tests/test_lifespan.py b/tests/test_lifespan.py
index 59b5a06b296..22a285623d0 100644
--- a/tests/test_lifespan.py
+++ b/tests/test_lifespan.py
@@ -81,7 +81,7 @@ def test_ticks_seen(self):
         assert self.df.steps.max() == 1
 
     def test_agent_lifetime(self):
-        lifetimes = self.df.groupby(["AgentID"]).Step.agg({"Step":
+        lifetimes = self.df.groupby(["AgentID"]).agg({"Step":
                                                            lambda x: len(x)})
         assert lifetimes.Step.max() == 2