diff --git a/.travis.yml b/.travis.yml
index 5800dc7..bb34d63 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -11,10 +11,10 @@ addons:
 install:
 - pip install --upgrade pip setuptools
 - pip install --upgrade sphinx coveralls
-- pip install --upgrade graphviz matplotlib
+- pip install --upgrade graphviz matplotlib wurlitzer
 - if [ "v$TRAVIS_PYTHON_VERSION" == "v2.7" ];
   then
-    pip install --upgrade futures mock;
+    pip install --upgrade mock;
   fi
 script:
 - coverage run setup.py test
diff --git a/setup.py b/setup.py
index e56047c..c92999b 100755
--- a/setup.py
+++ b/setup.py
@@ -42,7 +42,7 @@ def py2_only(*dependencies):
         "Programming Language :: Python :: 3.6",
         "Topic :: Software Development",
     ],
-    install_requires=["graphviz", "matplotlib"],
+    install_requires=["graphviz", "matplotlib", "wurlitzer"],
     tests_require=py2_only("mock"),
     test_suite="ActionTree.tests",
     use_2to3=True,