diff --git a/setup.py b/setup.py
index 79487b7..ceb546a 100644
--- a/setup.py
+++ b/setup.py
@@ -19,5 +19,5 @@
     license="Apache 2.0",
     packages=["bertviz"],
     include_package_data=True,
-    install_requires=["transformers>=2.0", "torch>=1.0", "tqdm", "boto3", "requests", "regex", "sentencepiece"],
-)
\ No newline at end of file
+    install_requires=["transformers>=2.0", "torch>=1.0", "tqdm", "boto3", "requests", "regex", "sentencepiece", "IPython"],
+)